回答編集履歴
1
追記
answer
CHANGED
|
@@ -1,4 +1,15 @@
|
|
|
1
1
|
```plain
|
|
2
2
|
docker run --gpus all --rm nvidia/cuda nvidia-smi
|
|
3
3
|
```
|
|
4
|
-
こちらは **nvidia-smiが実行できるか?** を確認するだけのコマンド例ではないでしょうか。
|
|
4
|
+
こちらは **nvidia-smiが実行できるか?** を確認するだけのコマンド例ではないでしょうか。
|
|
5
|
+
|
|
6
|
+
|
|
7
|
+
----
|
|
8
|
+
tensorflow/tensorflow:latest-gpu を使うならcudaは入ると思いますが。
|
|
9
|
+
|
|
10
|
+
[https://hub.docker.com/layers/tensorflow/tensorflow/latest-gpu/images/sha256-0cb24474909c8ef0a3772c64a0fd1cf4e5ff2b806d39fd36abf716d6ea7eefb3?context=explore](https://hub.docker.com/layers/tensorflow/tensorflow/latest-gpu/images/sha256-0cb24474909c8ef0a3772c64a0fd1cf4e5ff2b806d39fd36abf716d6ea7eefb3?context=explore)
|
|
11
|
+
の9が
|
|
12
|
+
```plain
|
|
13
|
+
RUN /bin/sh -c apt-get update && apt-get install -y --no-install-recommends cuda-cudart-11-2=11.2.146-1 cuda-compat-11-2 && ln -s cuda-11.2 /usr/local/cuda && rm -rf /var/lib/apt/lists/* # buildkit
|
|
14
|
+
```
|
|
15
|
+
となっています。
|