mirrors/LLM4Decompile
2
0
Fork 0
mirror of https://github.com/albertan017/LLM4Decompile.git synced 2026-06-17 01:55:50 +00:00
Code Issues Projects Releases Packages Wiki Activity

build: update pytorch and cuda versions in dockerfile

Browse source
This commit is contained in:
hanxen 2025-08-22 13:06:19 +09:00
commit 473973cf79
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
Dockerfile
View file

@ -1,5 +1,5 @@
# PyTorch base image with CUDA, cuDNN, and PyTorch preinstalled
FROM pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime
FROM pytorch/pytorch:2.7.1-cuda12.8-cudnn9-runtime
# Set working directory
WORKDIR /app