Skip to content

Commit

Permalink
deps-cuda: make sure cudnn gets installed, pin torch version for CUDA 11
Browse files Browse the repository at this point in the history
(sry, must commit to master directly – only Docker deploy is affected; current version is broken for all CUDA images)
  • Loading branch information
bertsky authored Oct 15, 2024
1 parent 85bde15 commit 6d8539c
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions Makefile
Original file line number Diff line number Diff line change
Expand Up @@ -97,7 +97,7 @@ deps-cuda:
# works, too:
shopt -s nullglob; \
$(PIP) install nvidia-pyindex \
&& $(PIP) install nvidia-cudnn-cu11==8.7.* \
&& $(PIP) install nvidia-cudnn-cu11~=8.7 \
nvidia-cublas-cu11~=11.11 \
nvidia-cusparse-cu11~=11.7 \
nvidia-cusolver-cu11~=11.4 \
Expand Down Expand Up @@ -158,7 +158,7 @@ deps-tf2:
fi

deps-torch:
$(PIP) install -i https://download.pytorch.org/whl/cu118 torch torchvision
$(PIP) install -i https://download.pytorch.org/whl/cu118 torchvision==0.16.2+cu118 torch==2.1.2+cu118

# Dependencies for deployment in an ubuntu/debian linux
deps-ubuntu:
Expand Down

0 comments on commit 6d8539c

Please sign in to comment.