Ubuntu 搭建 NVIDIA驱动 CUDA 和 pytorch GPU 环境

首先，查看自己的驱动型号

输入lspci，如上图所示，我的是Quadro K4200

搜索适合的驱动，并下载

记住存放的路径

 

接下来安装驱动：

参考：https://blog.youkuaiyun.com/kellyroslyn/article/details/91448848

1）删除、禁用驱动

删除旧的

sudo apt-get purge nvidia*

禁止自带的nouveau nvidia驱动

在/etc/modprobe.d/目录中找到黑名单

添加禁用 信息

blacklist nouveau 

options nouveau modeset=0 

保存退出

更新

重启电脑

 

重启之后独显没驱动了，界面可能变得很丑，但是不影响使用

进入文本输入模式，按ALT+CTRL+F1

输入Ubuntu的账号，密码

然后，设置权限，找到run存在的位置，并运行驱动安装程序

 

会自动进入安装，界面操作

安装结束输入输入

sudo nvidia-smi

 

 

二、接下来安装anaconda，这一步很简单

参考：https://blog.youkuaiyun.com/kellyroslyn/article/details/91402195

三、安装CUDA

登录官网：https://developer.nvidia.com/cuda-90-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1604&target_type=deblocal

 

下载到指定文件夹后，运行

一下几条命令

第二步的时候需要去路径中找自己的版本

 

 

 

 

按顺序操作，运行结束

三、安装pytorch

https://pytorch.org

下滑选择对应版本，python的版本可以再conda环境下输入python查看

下载太慢的话，参考：https://blog.youkuaiyun.com/qq_40971025/article/details/101448368

安装成功

输入python

安装成功