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