Windows下实现pytorch环境搭建

时间:2021-05-23

Anaconda安装

安装步骤:

1、官网下载安装包:https:///rdp/cudnn-download
需要注册并登录账号(这个需要一段时间),然后选择相对应的版本下载。


我的是CUDA10.2,所以我选择的是第二个。等待下载完成后进行解压,得到一个cuda文件夹,进入之后,全选,复制到之前CUDA所安装的文件夹下,有重复的进行替换即可。


查看cuDNN是否安装成功:

步骤如下:(进入安装的路径)

出现如下Result = PASS 说明cuDNN安装成功。


可以再接着执行deviceQuery.exe,如果出现Result = PASS 说明CUDA和cuDNN都已经安装成功了。

Pytorch安装

1、先配置torch环境。

cmd中输入:

conda create –n new python=3.7.4

(这儿python=xxx,大家可以先通过在cmd中直接输入python,就会出现你的python的版本,这儿的new是你所创建的虚拟环境的名称,大家可以自己取)
然后激活环境:
cmd中输入:

activate new

2、pytorch的安装

法一:打开pytorch官网:
https://pytorch.org/get-started/previous-versions/
找到自己所对应版本的pip即可(会慢些)。

法二:进入如下网址:https://download.pytorch.org/whl/torch_stable.html,下载自己所需要的torch和torchvision。
可以按快捷键ctrl+F进行搜索。


(cu102表示的是cuda10.2,大家根据自己cuda的版本进行搜索,cp37是python的版本,我的是3.7.4,win就是windows系统了,选择最新的版本下载即可)
查找torchvision也是同理。下载完成后,
要先进入刚才创建的torch环境中,即输入 activate new。
前面的(bash)就会变成 (new)。
接下来操作如下:


等待安装完成。

查看pytorch环境是否安装成功:


要记得先进入之前创建的环境中,然后依次输入蓝色方格中的代码,没有报错,且最后输出True,则说明pytorch环境安装完成。

到此这篇关于pytorch环境搭建(Windows)的文章就介绍到这了,更多相关pytorch环境搭建(Windows)内容请搜索以前的文章或继续浏览下面的相关文章希望大家以后多多支持!

声明:本页内容来源网络,仅供用户参考;我单位不保证亦不表示资料全面及准确无误,也不保证亦不表示这些资料为最新信息,如因任何原因,本网内容或者用户因倚赖本网内容造成任何损失或损害,我单位将不会负任何法律责任。如涉及版权问题,请提交至online#300.cn邮箱联系删除。

相关文章