Update Dockerfile
Browse files- 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 |
|