利用Anaconda安装pytorch搭建深度学习环境

利用Anaconda安装pytorch搭建深度学习环境,第1张

利用Anaconda安装pytorch搭建深度学习环境

一、NVIDIA驱动安装

 首先查看电脑的显卡版本,步骤为:此电脑右击-->管理-->设备管理器-->显示适配器。就可以看到电脑显卡的版本了。

得知以上的信息以后我们就可以对应我们的显卡去英伟达官网上去找相对应的显卡驱动更新或者下载了。显卡驱动的下载网址NVIDIA 驱动程序下载https://www.nvidia.cn/Download/index.aspx?lang=cn

然后根据自己的电脑的配置去选择驱动。这里Notebooks是笔记本的意思,所以如果你是笔记本电脑,那么产品系列那个选项就要选择(Notebooks)的。还有那个下载类型有两种一个是Studio版本,一个是Game Ready版本。其实两个版本都差不多,一个是偏办公用,一个是偏游戏娱乐。按如下 *** 作将驱动下载下来。然后点击下载来的程序,不断的下一步就好了。

安装好了显卡驱动以后。按下win+R组合键,打开cmd命令窗口。输入命令nvidia-smi。得到如下图的信息图,可以看到驱动的版本是496.76;最高支持的CUDA版本是11.5版本。得到显卡的最高支持的CUDA版本,我们就可以根据这个信息来安装环境了。

二、Anaconda 的安装

打开Anaconda | Individual Editionhttps://www.anaconda.com/products/individual下载最新版本的anaconda

若想下载其他版本,到清华镜像https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/选择对应的版本anaconda python 版本对应关系_茶佬牛逼-CSDN博客_anaconda python版本https://blog.csdn.net/yuejisuo1948/article/details/81043823

双击下载好的anaconda安装包,just me是说只供当前用户使用。all user 是供使用这台电脑的所有用户使用,是权限问题。对空间影响不大。如果你的电脑上只有建了一个用户,all users和just me 的作用是一样的,点击just me就好了。

     然后点击next,当让你选择安装安装路径的时候,一定不要选择默认安装位置,因为默认位置是c盘,以后要在anaconda里面创建环境的时候会很占内存,最好在D盘中创建一个文件夹来放anaconda。

 和图中一样将图中的√勾上,虽然出现红色的警告,但是要勾上,将anaconda添加到环境变量中去。然后点击完成就好了。

    安装完成以后,按下开始键在左边就会出现anaconda3这个文件夹,可以发现anaconda已经安装好了。

三、Pytorch环境安装

按下开始键,点击如图中的图标。打开anaconda的终端。

创建虚拟环境conda create -n 环境名字(英文) python=x.x(python版本),如下就是创建了一个名字叫pytorch,python是3.9版本的环境。

conda create -n pytorch python=3.9

执行如上的命令,就会创建一个新的虚拟环境,这个虚拟环境会安装一些基础的包,如下图所示。询问是否安装的时候,输入y。就可以创建环境了。若环境创建失败,打开.condarc将里面清华镜像全部删除:​​

当安装好了以后,执行conda env list这个命令,就可以看到多了一个pytorch这个环境。执行命令conda activate pytorch,激活这个环境。

安装pytorch-gup版的环境,由于pytorch的官网在国外,下载相关的环境包是比较慢的,所以我们给环境换源。在pytorch环境下执行如下的命名给环境换清华源。

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

conda config --set show_channel_urls yes

然后打开pytorch的官网,由于开头我们通过驱动检测到我的显卡为 GTX1050,最高支持cuda11.5版本,所以我们选择cuda11.1版本的cuda,然后将下面红色框框中的内容复制下来将复制的内容粘贴到pytorch环境下的终端,运行就可以开始下载环境所需要的依赖包了。

若安装过程中出现下面情况

可将.condarc文件中的https全部改成http,再重新安装一次。

五、验证CUDA和cudnn版本

  打开pycharm,创建一个新的工程,这时候我们就要选择我们在anaconda里面安装的环境,在界面的右下角,按照如图中的选项选择我们在anaconda中创建的深度学习环境,可以看到有pytorch环境,我们先选择pytorch环境。

此时刚刚的右下角已经有了我们刚刚选择的pytorch环境中的python了。我们在创建的工程里面创建一个python脚本,在脚本中运行如下代码,查看是否anconda在安装pytorch环境的时候也安装了cuda和cudnn。

import torch
print(torch.cuda.is_available())
print(torch.backends.cudnn.is_available())
print(torch.cuda_version)
print(torch.backends.cudnn.version())

运行之后出现下面结果就安装成功啦! 

 

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/zaji/5721327.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-12-18
下一篇 2022-12-18

发表评论

登录后才能评论

评论列表(0条)

保存