Update Dockerfile
Browse files- Dockerfile +2 -1
Dockerfile
CHANGED
|
@@ -1,4 +1,5 @@
|
|
| 1 |
-
FROM runpod/pytorch:2.2.1-py3.10-cuda12.1.1-devel-ubuntu22.04
|
|
|
|
| 2 |
WORKDIR /content
|
| 3 |
ENV PATH="/home/zebraslive/.local/bin:${PATH}"
|
| 4 |
|
|
|
|
| 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 |
|