1、查看是否安裝nvidia驅動:sudo lshw -c video
(configuration: driver=nvidia latency=0)
2、查看NVIDIA內核:cat /proc/driver/nvidia/version
查看NVIDIA驅動版本:sudo dpkg --list | grep nvidia-*
3、安裝NVDIA驅動:sudo apt install nvidia-driver-465
必須重啟:reboot
4、卸載NVDIA驅動:
sudo apt-get --purge remove nvidia*
sudo apt-get --purge remove "*cublas*" "cuda*"
sudo apt-get --purge remove "*nvidia*"
sudo rm -rf /usr/local/cuda-xx.x
5、cuda下載:
(root環境下裝好各個包,其他用戶增加配置即可vim ~/.bashrc)
https://developer.nvidia.com/cuda-toolkit-archive
修改配置:
sudo vim ~/.bashrc
最后添加:
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
保存:
source ~/.bashrc
測試:
cd /usr/local/cuda/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
查看版本:
cat /usr/local/cuda/version.txt或nvcc -V
卸載:
sudo /usr/local/cuda-10.2/bin/cuda-uninstaller
rm -rf /usr/local/cuda-10.2/
6、cudnn下載:
(安裝多個環境:環境配置里切換cuda,cudnn則是直接重裝,復制文件夾)
https://developer.nvidia.com/rdp/cudnn-archive#a-collapse820-111
tar zxvf cudnn-10.0-linux-x64-v7.6.5.32.tgz -C ./
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
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
測試:
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
7、安裝anaconda
https://www.anaconda.com/products/individual-b
bash Anaconda2-2019.07-Linux-x86_64.sh
重開終端
conda env list
8、卸載anaconda
rm -rf anaconda3 #文件夾名
(vi ~/.bashrc 刪除相關語句)
9、創建python37環境
conda create -n py37 python=3.7
source activate py37
刪除環境:conda remove -n py37 --all
10、安裝pytorch
https://pytorch.org/get-started/previous-versions/
pip3 install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
使用pip安裝 pip install torch==1.8.0
