Spaces:
Runtime error
Runtime error
Adding deps
Browse files- Dockerfile +2 -2
Dockerfile
CHANGED
@@ -12,12 +12,12 @@ COPY ./environment_docker.yml /code/environment_docker.yml
|
|
12 |
RUN conda env create -f /code/environment_docker.yml
|
13 |
|
14 |
# Activate the environment and install torch, torchvision, and torchaudio
|
15 |
-
RUN /
|
16 |
pytorch==2.1.2 \
|
17 |
torchvision==0.16.2 \
|
18 |
torchaudio==2.1.2 \
|
19 |
pytorch-cuda==11.8 \
|
20 |
-
cudatoolkit==11.8
|
21 |
|
22 |
COPY ./install_deps.sh /code/install_deps.sh
|
23 |
|
|
|
12 |
RUN conda env create -f /code/environment_docker.yml
|
13 |
|
14 |
# Activate the environment and install torch, torchvision, and torchaudio
|
15 |
+
RUN /bin/bash -c "source activate masaenv && conda install -n masaenv -c pytorch -c nvidia -c conda-forge \
|
16 |
pytorch==2.1.2 \
|
17 |
torchvision==0.16.2 \
|
18 |
torchaudio==2.1.2 \
|
19 |
pytorch-cuda==11.8 \
|
20 |
+
cudatoolkit==11.8"
|
21 |
|
22 |
COPY ./install_deps.sh /code/install_deps.sh
|
23 |
|