meepmoo commited on
Commit
66a37a5
·
verified ·
1 Parent(s): a441236

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -1
Dockerfile CHANGED
@@ -1,5 +1,6 @@
1
  #FROM runpod/pytorch:2.2.1-py3.10-cuda12.1.1-devel-ubuntu22.04
2
- FROM mybigpai-public-registry.cn-beijing.cr.aliyuncs.com/easycv/torch_cuda:cogvideox_fun
 
3
  WORKDIR /content
4
  ENV PATH="/home/zebraslive/.local/bin:${PATH}"
5
 
 
1
  #FROM runpod/pytorch:2.2.1-py3.10-cuda12.1.1-devel-ubuntu22.04
2
+ #FROM mybigpai-public-registry.cn-beijing.cr.aliyuncs.com/easycv/torch_cuda:cogvideox_fun
3
+ FROM nvidia/cuda:11.8.0-devel-ubuntu22.04
4
  WORKDIR /content
5
  ENV PATH="/home/zebraslive/.local/bin:${PATH}"
6