Skip to content

Commit

Permalink
Revert back
Browse files Browse the repository at this point in the history
  • Loading branch information
@georgeyiasemis
georgeyiasemis committed Jun 23, 2024
1 parent 8b9c4c6 commit e41f3b3
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions docker/Dockerfile
View file
Original file line number Diff line number Diff line change
@@ -1,9 +1,9 @@
ARG CUDA="11.3.0"
ARG PYTORCH="1.10"
ARG PYTHON="3.10"
ARG PYTHON="3.8"

# TODO: conda installs its own version of cuda
FROM nvidia/cuda:${CUDA}-devel-ubuntu20.04
FROM nvidia/cuda:${CUDA}-devel-ubuntu18.04

ENV CUDA_PATH /usr/local/cuda
ENV CUDA_ROOT /usr/local/cuda/bin
Expand Down

0 comments on commit e41f3b3

Please sign in to comment.