Update setup script

dinhanhx · Aug 17, 2021 · f926542 · f926542
1 parent 03ad70b
commit f926542
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/UNITER/conda/setup_uniter.sh b/UNITER/conda/setup_uniter.sh
@@ -6,9 +6,9 @@ cd ../
 conda install pytorch==1.0.1 torchvision==0.2.2 cudatoolkit=10.2 -c pytorch --yes
 conda install scikit-learn --yes
 pip install pytorch-pretrained-bert==0.6.2 tensorboardX==1.7 ipdb==0.12 lz4==2.1.9 lmdb==0.97
-pip install toolz cytoolz msgpack msgpack-numpy
+pip install toolz cytoolz msgpack msgpack-numpy cmake
 
-HOROVOD_WITH_PYTORCH=1 pip install --no-cache-dir horovod
+HOROVOD_WITH_PYTORCH=1 PATH=$PATH pip install --no-cache-dir horovod
 
 # apex (gcc >= 5.4.0 required)
 git clone https://github.com/jackroos/apex