ICode9

精准搜索请尝试: 精确搜索
首页 > 系统相关> 文章详细

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

2022-02-03 13:00:32  阅读:190  来源: 互联网

标签:sudo 495.46 apt Ubuntu Pytorch 64 NVIDIA 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

第五步,验证环境

标签:sudo,495.46,apt,Ubuntu,Pytorch,64,NVIDIA,GPU,安装
来源: https://blog.csdn.net/harryxia2014/article/details/122775094

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有