ubuntu安装cuda+cudnn+pytorch

目录

1、修改分辨率

2、安装 gcc/cmake

3、安装驱动

3.1 下一个最新的

3.2 禁用Nouveau驱动

3.3 关闭显示模式

4、安装CUDA

4.1 下载对应CUDA版本

4.2 安装

 4.3 配置环境变量

4.4测试安装是否成功

5、安装cuDNN

5.1、下载cuDNN

cuDNN Archive | NVIDIA Developer5.2、解压

5.3、配置

6、重装Python

6.1、卸载Python

6.2、安装编译依赖项

6.3、下载源码

6.4、编译

6.5、 设置软链接

6.6、 pip问题

7、安装pytorch

7.1安装

7.2 测试

---

假定ubuntu已经安装好了

1、修改分辨率

#sudo gedit /etc/default/grub

GRUB_GFXMODE=1024x768

sudo update-grub
#重启

2、安装 gcc/cmake

apt-get install build-essential

apt-get install cmake

检查是否安装成功：

gcc --version

cmake –version

有版本号显示说明安装成功

3、安装驱动

3.1 下一个最新的

NVIDIA 驱动程序下载

3.2 禁用Nouveau驱动

 sudo gedit /etc/modprobe.d/blacklist-nouveau.conf 
 #在文件 blacklist-nouveau.conf 中加入如下内容：
  blacklist nouveau
  blacklist lbm-nouveau
  options nouveau modeset=0
  alias nouveau off
  alias lbm-nouveau off
  #保存 

#禁用nouveau 内核模块
echo options nouveau modeset=0 | sudo tee -a /etc/modprobe.d/nouveau-kms.conf
sudo update-initramfs -u
reboot #重启
lsmod |grep nouveau #无显示则成功 当然驱动没了你的桌面分辨率会比较大。

3.3 关闭显示模式

 

#进入tty 终端显示Ctrl+Alt+F1，退出Ctrl+Alt+F7
#登录
service lightdm stop #关闭显示管理
cd /home/username/Downloads #安装驱动；电脑的名字
sh NVIDIA-Linux-x86_64-415.27.run --no-opengl-files #安装；对应下在的驱动文件
# 一路默认
# 重启
# 输入nvidia-smi 有输出则成功。

4、安装CUDA

请先参考pytorch对应的CUDA版本

PyTorch

4.1 下载对应CUDA版本

CUDA Toolkit Archive | NVIDIA Developer

4.2 安装

sh cuda_10.0.130_410.48_linux.run

 4.3 配置环境变量

sudo gedit ~/.bashrc //末尾添加 
export CUDA_HOME=/usr/local/cuda-10.0 
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64:$LD_LIBRARY_PATH 
export PATH=/usr/local/cuda-10.0/bin:$PATH

4.4测试安装是否成功

cd /usr/local/cuda-9.0/samples/1_Utilities/deviceQuery

make

sudo ./deviceQuery

5、安装cuDNN

5.1、下载cuDNN

cuDNN Archive | NVIDIA Developer
5.2、解压

tar -zxvf cudnn-10.0-linux-x64-v7.5.0.56.tgz

5.3、配置

将cuda/include/cudnn.h和cuda/lib64/libcudnn*拷贝到之前安装的cuda的文件夹下：

cp cuda/include/cudnn.h  /usr/local/cuda/include

cp cuda/lib64/libcudnn*  /usr/local/cuda/lib64

6、重装Python

6.1、卸载Python

sudo apt-get remove python
sudo rm -r /etc/py*
sudo rm -r /usr/bin/py*
sudo rm -r /usr/lib/py*
sudo rm -r /usr/local/lib/py*
sudo rm -r /usr/share/py*
sudo rm -r /usr/share/man/man1/py*

6.2、安装编译依赖项

sudo apt install -y wget build-essential li

6.3、下载源码

wget https://www.python.org/ftp/python/3.9.0/Python-3.9.0b4.tgz
 
tar -zxvf Python-3.9.0b4.tgz # 解压源码包

6.4、编译

#编译参数设置
./configure --prefix=/usr/local/python3
 
#编译
make
 
#安装
sudo make install

6.5、 设置软链接

#查看，出错的就删掉再改
ll /usr/bin | grep python
ll /usr/bin | grep pip

sudo rm python
sudo rm python3 #并不会删除 python2.7 和 python3.5
 
sudo ln -s /usr/local/python3/bin/python3.9 /usr/bin/python3
sudo ln -s /usr/local/python3/bin/python3.9 /usr/bin/python
 
#为 pip 设置软链接
sudo ln -s /usr/local/python3/bin/pip3.9 /usr/bin/pip3
sudo ln -s /usr/local/python3/bin/pip3.9 /usr/bin/pip

6.6、 pip问题

subprocess.CalledProcessError: Command '('lsb_release', '-a')' returned non-zero exit status 

sudo rm /usr/bin/lsb_release

7、安装pytorch

7.1安装

PyTorch

 网络有问题就手动下载

7.2 测试

import 	torch
print(torch.cuda.is_available())

8、修改pip源

mkdir ~/.pip
vim ~/.pip/pip.conf

[global]
index-url = http://pypi.douban.com/simple
trusted-host = pypi.douban.com