install gpu

cm3apps/install.txt

@@ -495,9 +495,40 @@ D) Install Torch
         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*
+           sudo apt remove --autoremove nvidia-cuda-toolkit
+           sudo apt remove --autoremove nvidia-*
+           2) purge
+           sudo apt-get purge nvidia*
+           sudo apt-get autoremove
+           sudo apt-get autoclean
+           sudo rm -rf /usr/local/cuda*
+           3) install graphics
+           sudo apt update
+           sudo add-apt-repository ppa:graphics-drivers
+           sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub
+           sudo bash -c 'echo "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64 /" > /etc/apt/sources.list.d/cuda.list'
+           sudo bash -c 'echo "deb http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu2004/x86_64 /" > /etc/apt/sources.list.d/cuda_learn.list'
+           sudo apt update
+           sudo apt install cuda-11-0
+           sudo apt install libcudnn7
+           add in your .bashrc
+           export PATH=/usr/local/cuda/bin/:$PATH
+           export LD_LIBRARY_PATH=/usr/local/cuda/lib64/:$LD_LIBRARY_PATH
+           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/cu101/libtorch-shared-with-deps-1.7.0%2Bcu101.zip
+        wget https://download.pytorch.org/libtorch/cu110/libtorch-shared-with-deps-1.7.0%2Bcu110.zip
 
 
         add export TORCHDIR=$HOME/local/libtorch in your .bashrc