Your Name
commited on
Commit
•
3177c4d
1
Parent(s):
08794d3
update
Browse files- Dockerfile +4 -2
Dockerfile
CHANGED
|
@@ -1,4 +1,5 @@
|
|
| 1 |
-
FROM nvidia/cuda:11.1.1-devel-ubuntu20.04
|
|
|
|
| 2 |
|
| 3 |
ENV CUDA_HOME=/usr/local/cuda
|
| 4 |
ENV PATH=${CUDA_HOME}/bin:/home/${USER_NAME}/.local/bin:${PATH}
|
|
@@ -23,7 +24,8 @@ RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-ins
|
|
| 23 |
wget \
|
| 24 |
&& rm -rf /var/lib/apt/lists/*
|
| 25 |
|
| 26 |
-
RUN pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html
|
|
|
|
| 27 |
|
| 28 |
|
| 29 |
WORKDIR /code
|
|
|
|
| 1 |
+
# FROM nvidia/cuda:11.1.1-devel-ubuntu20.04
|
| 2 |
+
FROM nvidia/cuda:11.8.0-devel-ubuntu22.04
|
| 3 |
|
| 4 |
ENV CUDA_HOME=/usr/local/cuda
|
| 5 |
ENV PATH=${CUDA_HOME}/bin:/home/${USER_NAME}/.local/bin:${PATH}
|
|
|
|
| 24 |
wget \
|
| 25 |
&& rm -rf /var/lib/apt/lists/*
|
| 26 |
|
| 27 |
+
# RUN pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html
|
| 28 |
+
RUN pip install torch==2.0.0 torchvision==0.15.1 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118
|
| 29 |
|
| 30 |
|
| 31 |
WORKDIR /code
|