Hugging Face's logo

tezuesh
/
IBLlama_v1

Any-to-Any
omega
omegalabs
bittensor
agi
Model card Files Community
tezuesh commited on
Commit
d00ea17
·
verified ·
1 Parent(s): 3687204

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -1,6 +1,6 @@
1
  # FROM nvidia/cuda:12.1.0-cudnn8-devel-ubuntu22.04
2
- # FROM nvidia/cuda:12.1-cudnn8-devel-ubuntu22.04
3
- FROM nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04 # Uses CUDA 12.1.1 + cuDNN 8.9.x
4
 
5
 
6
  # Set environment variables
 
1
  # FROM nvidia/cuda:12.1.0-cudnn8-devel-ubuntu22.04
2
+ FROM nvidia/cuda:12.3.2-cudnn9-devel-ubuntu22.04
3
+
4
 
5
 
6
  # Set environment variables