SW777SW commited on
Commit
f949c0e
·
verified ·
1 Parent(s): 5e7d472

Update Dockerfile

Browse files
Files changed (1) hide show
  1. 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=12.1 -c pytorch -c nvidia -y && \
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 && \