!sudo apt install cuda-toolkit-11-2 !wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-ubuntu1604.pin !sudo mv cuda-ubuntu1604.pin /etc/apt/preferences.d/cuda-repository-pin-600 !sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub !sudo add-apt-repository "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/ /" !sudo apt-get update !sudo apt-get -y install cuda !sudo apt --fix-broken install !sudo apt install cuda-cudart-11-2 import os os.environ['PATH'] = '/usr/local/cuda-11.2/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/tools/node/bin:/tools/google-cloud-sdk/bin:/opt/bin' os.environ['LD_LIBRARY_PATH'] = '/usr/local/cuda-11.2/lib64:/usr/local/cuda-11.2/lib64:/usr/lib64-nvidia' !cp "/content/drive/My Drive/cudnn-11.2-linux-x64-v8.1.0.0.tgz" "/content/cudnn-11.2.tgz" !mkdir cudnn !tar -C cudnn/ -zxvf cudnn-11.2.tgz %cd cudnn/ !sudo cp cuda/include/cudnn*.h /usr/local/cuda-11.2/include !sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.2/lib64 !sudo chmod a+r /usr/local/cuda-11.2/include/cudnn*.h /usr/local/cuda-11.2/lib64/libcudnn* !sudo ldconfig %cd /content/
! nvcc --version !nvidia-smi import chainer print("-----------------SYSTEM INFO----------------------") chainer.print_runtime_info()
var
This content, along with any associated source code and files, is licensed under The Code Project Open License (CPOL)