Ubuntu16安装nvidia驱动与cuda10.2 pytorch1.6（亲测可用）

pytorch更新了好多东西， 我还停留在pytorch1.1, 这里phoenix倒腾了一下电脑， 重装了cuda12与pytorch1.6

驱动以及cuda下载地址：

cuda：https://developer.nvidia.com/cuda-toolkit-archive

nvidia driver https://www.nvidia.cn/Download/Find.aspx?lang=cn

这里先选择cuda，然后按照对应的cuda后边的数字选择对应的驱动版本,， 例如我安装的cuda：cuda_10.2.89_440.33.01_linux.run， 这里就选择对应于440.33.01的显卡驱动， 如果找不到，那么驱动的数字最后的33.01可以不一样， 最好比它高一点。 应该说显卡驱动只要比440高的都行， 我没有试过

1. 卸载原有驱动

sudo apt remove nvidia-*
sudo apt autoremove

2. 安装显卡驱动

按下CTRL+ALT+F1， 进入命令行界面

sudo service lightdm stop  #禁用图形界面
sudo chmod +x NVIDIA-Linux-x86_64-440.33.01.run  # 赋予可执行权限
​
# -no-opengl-files这个参数可以放置循环启动， 不过有时候不加它倒是正常，加上了反而循环启动
sudo sh NVIDIA-Linux-x86_64-440.33.01.run -no-x-check -no-nouveau-check -no-opengl-files  # 安装
​
# 测试
nvidia-smi
​
sudo service lightdm start #启用图形界面 

3. 卸载原有cuda

sudo /usr/local/cuda-10.0/bin/uninstall_cuda_10.0.pl
​
sudo rm -rf /usr/local/cuda-10.0

4. 安装cuda

sudo chmod +x cuda_10.2.89_440.33.01_linux.run
​
sh cuda_10.2.89_440.33.01_linux.run
​
# accept

可能会出现以下的东西：

CUDA Installer
​
-[X] driver
-[X]CUDA ToolKit
-[X]CUDA Samples
-[X]CUDA Demo Suit
-[X]CUDA Documentation
Options
Install

由于之前已经安装过显卡驱动， 这里选中driver，点击回车将X号去掉， 然后选择Install回车即可正常安装

5. 安装pytorch

打开官网直接安装

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch