1200字范文,内容丰富有趣,写作的好帮手!
1200字范文 > Ubuntu20.04下NVIDIA驱动+anaconda3+cuda+cudnn+pytorch安装

Ubuntu20.04下NVIDIA驱动+anaconda3+cuda+cudnn+pytorch安装

时间:2023-01-06 14:57:46

相关推荐

Ubuntu20.04下NVIDIA驱动+anaconda3+cuda+cudnn+pytorch安装

声明:此文章是在初步接触深度学习的时候,参考几位前辈的博客自己进行安装后整理的步骤,所参考博客链接如下:

#/ashome123/article/details/105822040#/qq_42779103/article/details/108754887#/weixin_44776894/article/details/106159483

正文开始——————————————————————————

目录

一、系统要求二、系统安装好后先安装显卡的驱动2.1查看是否有显卡驱动2.2显卡驱动安装2.3本地路径下安装驱动2.3.1安装之前先卸载已经存在的驱动2.3.2禁用nouveau2.3.32.3.4开始安装驱动2.3.5验证是否安装成功 三、安装Anaconda33.1 安装包的下载3.2安装3.3配置环境变量 四、安装cuda4.1下载cuda4.2安装方法14.2安装方法2(推荐)4.3配置环境变量4.3验证安装 五、安装cuDNN5.1下载5.2安装 六、安装pytorch6.1进入官网选择当前机器环境6.2运行command中的命令6.3清华镜像源安装6.4测试安装是否成功 七、总结与指正

一、系统要求

Ubuntu20.04

二、系统安装好后先安装显卡的驱动

2.1查看是否有显卡驱动

#在终端输入以下命令进行查看nvidia-smi

如果无法查看,就说明还未安装显卡驱动

注:查看推荐驱动命令

ubuntu-drivers devices

带有recommended的就是推荐的显卡驱动

2.2显卡驱动安装

在软件和更新中,选择附加驱动,选择上一步查看到的驱动版本点击附加驱动等待安装就可以了,安装完成后重启

2.3本地路径下安装驱动

2.3.1安装之前先卸载已经存在的驱动

sudo apt-get remove --purge nvidia*

2.3.2禁用nouveau

通过以下命令打开/etc/modprobe.d/blacklist-nouveau.conf文件

sudo gedit /etc/modprobe.d/blacklist.conf

在文件最后添加

blacklist nouveau

保存后执行如下命令禁用nouveau

sudo update-initramfs -u

2.3.3

输入以下命令关闭桌面服务后即可开始安装驱动

sudo service lightdm stop

2.3.4开始安装驱动

首先在官网下载自己所需要的驱动/Download/index.aspx?lang=cn

之后在下载目录下打开终端

运行刚下载的驱动程序

sudo ./***#***为你所下载的驱动的名称,要全部复制再运行

之后按照如下步骤安装即可:

2.3.5验证是否安装成功

在终端输入nvidia-smi

三、安装Anaconda3

3.1 安装包的下载

anaconda官方下载地址:

/products/individual

清华镜像源下载地址:

https://mirrors.tuna./anaconda/archive/

3.2安装

在下载目录下打开终端

bash Anaconda3-.02-Linux-x86_64.sh

一直回车加yes就可以完成安装了,这部分还是特别好操作的

3.3配置环境变量

在终端输入命令

gedit ~/.bashrc

在打开文件最后一行添加:

export PATH="/home/awei/anaconda3/bin:$PATH"

这里的路径选择是你的Anaconda的默认安装路径,默认在home下。

点击保存,关闭bashrc文件,最后一定要在终端输入下面命令才能生效:

source ~/.bashrc

输入以下命令即可打开相应

spyder #打开Spyderipython #打开IPythonjupyter notebook #打开Jupyter notebookanaconda-navigator #打开Anaconda Navigator

四、安装cuda

4.1下载cuda

在官网下载相对应的版本进行下载

#/?t=/cuda-downloads

4.2安装方法1

终端命令直接安装,无需下载

4.2安装方法2(推荐)

下载目录下运行终端

chmod +x cuda_11.0.3_450.51.06_linux.runsudo sh cuda_11.0.3_450.51.06_linux.run

选择Continue

输入accept

cuda安装包是自带显卡驱动的,因为在前面已经安装了驱动,所以按空格去掉安装显卡驱动的选项,然后选择install

4.3配置环境变量

输入gedit ~/.bashrc命令打开文件,在文件结尾输入以下语句

export PATH=/usr/local/cuda-10.2/bin${PATH:+:${PATH}}export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

更新环境变量配置source ~/.bashrc

4.3验证安装

输入nvcc -V命令查看cuda信息

五、安装cuDNN

5.1下载

下载链接

下载时需注册账号,按照提示操作即可

/cudnn-download-survey

5.2安装

进入下载目录,在终端打开,解压下载的包,并将其拷贝到lib64和include这两个目录

tar -zxf cudnn-11.0-linux-x64-v8.0.3.33.tgzcd cudasudo cp lib64/* /usr/local/cuda/lib64/sudo cp include/* /usr/local/cuda/include/

六、安装pytorch

6.1进入官网选择当前机器环境

/get-started/locally/

6.2运行command中的命令

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

按照提示向下操作即可

6.3清华镜像源安装

pytorch在国内的网络环境下会出现下载速度慢,或者下载失败的情况,可以添加清华镜像源下载

conda config --add channels https://mirrors.tuna./anaconda/cloud/pytorch/conda config --add channels https://mirrors.tuna./anaconda/pkgs/free/conda config --add channels https://mirrors.tuna./anaconda/pkgs/main/conda config --set show_channel_urls yes

配置完成后运行的command命令改为,这里删除了 -c pytorch

conda install pytorch torchvision cudatoolkit=10.2

按照提示输入y等待安装完成,如下图

6.4测试安装是否成功

在终端下依次输入以下命令

python #显示python版本import torchimport torchvisiontorch.cuda.is_available()

未报错则安装成功

七、总结与指正

刚入门不久,很多东西都在慢慢的学习中,因为自己是在安装完成后进行的总结,所以文章中的图片部分来自于网络,但所要表达的意思都完全的表达出来了。敬请各位指正。

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。