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

本文主要是介绍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-495.46.run --uninstall
将nouveau添加到黑名单，这里需要sudo权限：

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

blacklist nouveau

之后，手动下载NVIDIA驱动程序：

Download Drivers​www.nvidia.com/Download/index.aspx?lang=en-us正在上传…重新上传取消

下载NVIDIA-Linux-x86_64-495.46.run

执行命令：sh NVIDIA-Linux-x86_64-495.46.run

安装成功后

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

显示显卡信息

NVIDIA-SMI 495.46       Driver Version: 495.46       CUDA Version: 11.5  

3.2 安装CUDA

下载：cuda-repo-ubuntu2004-11-5-local_11.5.0-495.29.05-1_amd64.deb

执行命令：

sudo dpkg -i cuda-repo-ubuntu2004-11-5-local_11.5.0-495.29.05-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu2004-11-5-local/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda

第四步，安装Pytorch

执行命令：

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

第五步，验证环境

这篇关于Ubuntu 20.04-Pytorch-GPU系统搭建指南的文章就介绍到这儿，希望我们推荐的文章对大家有所帮助，也希望大家多多支持为之网！