Spaces:
Paused
Paused
Update Dockerfile
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
|
@@ -26,7 +26,7 @@ RUN conda update -n base -c defaults conda
|
|
| 26 |
RUN conda create -n instantsplat python=3.11 cmake=3.14.0 -y && \
|
| 27 |
echo "source activate instantsplat" > ~/.bashrc && \
|
| 28 |
/bin/bash -c "source activate instantsplat" && \
|
| 29 |
-
conda install pytorch torchvision pytorch-cuda=
|
| 30 |
cd InstantSplat && \
|
| 31 |
pip install -r requirements.txt && \
|
| 32 |
pip install submodules/simple-knn && \
|
|
|
|
| 26 |
RUN conda create -n instantsplat python=3.11 cmake=3.14.0 -y && \
|
| 27 |
echo "source activate instantsplat" > ~/.bashrc && \
|
| 28 |
/bin/bash -c "source activate instantsplat" && \
|
| 29 |
+
conda install pytorch torchvision pytorch-cuda=11.8 -c pytorch -c nvidia -y && \
|
| 30 |
cd InstantSplat && \
|
| 31 |
pip install -r requirements.txt && \
|
| 32 |
pip install submodules/simple-knn && \
|