Skip to content

Commit

Permalink
fix: change cuda to 12.1.0
Browse files Browse the repository at this point in the history
  • Loading branch information
@origyZ
origyZ committed May 10, 2024
1 parent f12cf5e commit 4afe54c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pytorch/base.Dockerfile
View file
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
FROM nvidia/cuda:12.2.2-devel-ubuntu20.04
FROM nvidia/cuda:12.1.0-devel-ubuntu20.04

RUN set -eux; \
apt-get update; \
Expand Down

0 comments on commit 4afe54c

Please sign in to comment.