ThomasBlumet
commited on
Commit
·
9e5775a
1
Parent(s):
e45d465
torch 2.0.1 with cuda 11.8 & correct cuda img
Browse files- Dockerfile +3 -1
Dockerfile
CHANGED
|
@@ -1,6 +1,8 @@
|
|
| 1 |
# For more information, please refer to https://aka.ms/vscode-docker-python
|
| 2 |
#FROM python:3.10-slim
|
| 3 |
-
FROM nvidia/cuda:
|
|
|
|
|
|
|
| 4 |
|
| 5 |
#ENV DEBIAN_FRONTEND=noninteractive
|
| 6 |
|
|
|
|
| 1 |
# For more information, please refer to https://aka.ms/vscode-docker-python
|
| 2 |
#FROM python:3.10-slim
|
| 3 |
+
FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
|
| 4 |
+
|
| 5 |
+
#nvidia/cuda:12.6.3-cudnn-devel-ubuntu24.04
|
| 6 |
|
| 7 |
#ENV DEBIAN_FRONTEND=noninteractive
|
| 8 |
|