Skip to content

Commit 49c5387

Browse files
committed
Change CUDA image to supported version
1 parent 7882e18 commit 49c5387

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

Dockerfiles/install-ubuntu.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ done
1717

1818
if $cuda; then
1919
# install libcupti
20-
apt-get -y install cuda-command-line-tools-11-7
20+
apt-get -y install cuda-command-line-tools-11-6
2121
fi
2222

2323
apt-get -y clean

Dockerfiles/ubuntu-cuda

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
FROM nvidia/cuda:11.7.0-cudnn8-devel-ubuntu22.04
1+
FROM nvidia/cuda:11.6.2-cudnn8-devel-ubuntu20.04
22

33
# copy the contents of this repository to the container
44
COPY . tensorflow_cc

0 commit comments

Comments
 (0)