cuda pytorch 环境变量_科学网—[深度学习]Ubuntu18.04配置深度学习环境Pytorch1.0+CUDA10+NVIDA驱动 - 王飞的博文...

Ubuntu18.04 Install Pytorch

安装NVIDA GPU显卡驱动

1. 查询显卡型号

2. 官网下载适合自己显卡的驱动.run文件

3. 卸载原有的NVIDIA驱动，sudo apt-get remove --purge nvidia*    sudo apt-get autoremove；如果是使用官网下载的.run驱动，找到安装文件sudo ./***.run --uninstall

4. 禁用nouveau驱动sudo gedit /etc/modprobe.d/blacklist.conf，在末尾添加:blacklist nouveau，然后执行：sudo update-initramfs -u，重启进入命令行界面后执行：lsmod | grep nouveau。如果没有屏幕输出，说明禁用nouveau成功。

5. 进入Ctrl+Alt+F3进入命令行模式，关闭gdm服务sudo /etc/init.d/gdm3 stop

6. 找到下载的驱动安装，sudo ./*****.sh -no-opengl-files(务必加上此参数)

7. 重启sudo reboot

安装CUDA10.1

1. 官网下载linux, x86_64, Ubuntu, 18.04(或17.04)的runfile

https://developer.nvidia.com/cuda-downloads

2. 安装

sudo sh cuda_10.0.130_410.48_linux.run

按空格读完声明

在安装过程中会询问是否安装显卡驱动，我们已经安装所以将驱动项去掉即可。

安装完成后可能会有警告，提示samplee