From 341f7f3f3c60ff685a6e40c9efde2aa87749590a Mon Sep 17 00:00:00 2001 From: ling <l.wu@ulg.ac.be> Date: Tue, 3 Nov 2020 12:34:19 +0100 Subject: [PATCH] install cudnn --- cm3apps/install.txt | 22 ++++++++++------------ 1 file changed, 10 insertions(+), 12 deletions(-) diff --git a/cm3apps/install.txt b/cm3apps/install.txt index 48b68a813..41516f33d 100644 --- a/cm3apps/install.txt +++ b/cm3apps/install.txt @@ -492,8 +492,8 @@ 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 + 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 1) clean up sudo rm /etc/apt/sources.list.d/cuda* @@ -519,16 +519,14 @@ D) Install Torch 4) install cudnn download cudnn from nvidia (you need to register and downloadmanualy https://developer.nvidia.com/compute/machine-learning/cudnn/secure/8.0.4/11.0_20200923/cudnn-11.0-linux-x64-v8.0.4.30.tgz - - - - -A -A - - sudo apt-get install nvidia-cuda-toolkit - //check version nvcc --version - wget https://download.pytorch.org/libtorch/cu110/libtorch-shared-with-deps-1.7.0%2Bcu110.zip + copy in your local directory + tar xvf cudnn-11.0-linux-x64-v8.0.4.30.tgz + sudo cp cuda/include/cudnn*.h /usr/local/cuda/include + sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64 + sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn* + 5) download torch with gpu + wget https://download.pytorch.org/libtorch/cu110/libtorch-shared-with-deps-1.7.0%2Bcu110.zip + unzip libtorch-shared-with-deps-1.7.0%2Bcu110.zip add export TORCHDIR=$HOME/local/libtorch in your .bashrc -- GitLab