diff --git a/cm3apps/install.txt b/cm3apps/install.txt index 22a84d0818f2c91c59a01f9e5d7658638df33b43..f19eb4404303ae653ab08a9453976550f9b57fee 100644 --- a/cm3apps/install.txt +++ b/cm3apps/install.txt @@ -491,8 +491,14 @@ C) Install Open Cascade (optional) D) Install Torch cd local + // no gpu wget https://download.pytorch.org/libtorch/cpu/libtorch-shared-with-deps-1.4.0%2Bcpu.zip unzip libtorch-shared-with-deps-1.4.0+cpu.zip + // gpu + sudo apt-get install nvidia-cuda-toolkit + //check version nvcc --version + wget https://download.pytorch.org/libtorch/cu101/libtorch-shared-with-deps-1.7.0%2Bcu101.zip + add export TORCHDIR=$HOME/local/libtorch in your .bashrc