Update install-deps.sh

lab4d-org · Apr 28, 2024 · 168f6fe · 168f6fe
1 parent fd9eb13
commit 168f6fe
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/scripts/install-deps.sh b/scripts/install-deps.sh
@@ -1,11 +1,11 @@
 mim install mmcv
 
-(cd lab4d/third_party/quaternion && pip install .)
+(cd lab4d/third_party/quaternion && CUDA_HOME=$CONDA_PREFIX pip install .)
 
 mkdir ./preprocess/third_party/Track-Anything/checkpoints; gdown --fuzzy https://drive.google.com/uc?id=10wGdKSUOie0XmCr8SQ2A2FeDe-mfn5w3 -O ./preprocess/third_party/Track-Anything/checkpoints/E2FGVI-HQ-CVPR22.pth
 
 wget https://www.dropbox.com/s/bgsodsnnbxdoza3/vcn_rob.pth -O ./preprocess/third_party/vcnplus/vcn_rob.pth
 
 wget https://www.dropbox.com/s/51cjzo8zgz966t5/human.pth -O preprocess/third_party/viewpoint/human.pth
 
-wget https://www.dropbox.com/s/1464pg6c9ce8rve/quad.pth -O preprocess/third_party/viewpoint/quad.pth
+wget https://www.dropbox.com/s/1464pg6c9ce8rve/quad.pth -O preprocess/third_party/viewpoint/quad.pth