Ubuntu 20.04-Pytorch-GPU系统环境搭建指南

本人硬件环境:

CPU:Intel Core i7 6700

GPU:NVIDIA GTX 1060 6G

内存:SAMSUNG DDR4 32GB

硬盘:双SSD 256G,双系统(windows,ubuntu)

第一步,安装Ubuntu.

1.1 准备工作

1.1.1 下载 Ubuntu 镜像

打开ubuntu 20.04的下载地址,进入页面后点击右边的Download按钮开始下载。

Ubuntu 20.04​ubuntu.com/download/desktop

你会得到一个Ubuntu ISO文件。

制作U盘启动盘然后安装即可。

第二步,配置Python环境.

下载Anaconda3-2021.11-Linux-x86_64.sh

执行命令:bash Anaconda3-2021.11-Linux-x86_64.sh

安装晚后restart

第三步,安装NVIDIA驱动,配置CUDA+

首先,建议你需要满足这两个条件,虽然没有任何证据证明与成功安装的相关性,但是这样做的确会减少报错:

  • BIOS务必关闭Secure Boot.
  • Python3版本应不低于3.5.

如果你按照上面的操作步骤,你只需要在BIOS里关闭Secure Boot。

3.1卸载已有的驱动程序,禁用nouveau

#卸载之前的所有驱动
#apt-get安装的原有驱动:
sudo apt-get remove --purge nvidia*
 
#runfile安装的原有驱动:
sudo chmod +x *.run
sudo ./NVIDIA-Linux-x86_64-510.47.03.run --uninstall
将nouveau添加到黑名单,这里需要sudo权限:

sudo gedit /etc/modprobe.d/blacklist.conf
最后一行添加:

blacklist nouveau

之后,手动下载NVIDIA驱动程序:

下载NVIDIA-Linux-x86_64-510.47.03.run

执行命令:sh NVIDIA-Linux-x86_64-510.47.03.run

安装成功后

验证驱动
执行命令:nvidia-smi

显示显卡信息

3.2 安装CUDA

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/11.6.0/local_installers/cuda-repo-ubuntu2004-11-6-local_11.6.0-510.39.01-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-11-6-local_11.6.0-510.39.01-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu2004-11-6-local/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda

第四步,安装Pytorch

下载:torch-1.9.0+cu102-cp39-cp39-linux_x86_64.whl

torchaudio-0.9.0-cp39-cp39-linux_x86_64.whl

torchvision-0.10.0+cu111-cp39-cp39-linux_x86_64.whl

执行命令:

pip install torch-1.9.0+cu102-cp39-cp39-linux_x86_64.whl

pip install torchaudio-0.9.0-cp39-cp39-linux_x86_64.whl

pip install torchvision-0.10.0+cu111-cp39-cp39-linux_x86_64.whl

第五步,验证环境

>>> import torch
>>> torch.cuda.is_available()
True

结果是true就ok了。如下图

猜你喜欢

转载自blog.csdn.net/harryxia2014/article/details/122775094