本地安装pytorch-超级详细

1.查看自己是否有GPU

打开控制面板,如果能够找到NVIDIA,就说明你有GPU
在这里插入图片描述

2.查看CUDA型号

打开NVIDIA控制面板,点击系统信息。
在这里插入图片描述
点击组件,NVCUDA64.DLL后边对应的数字就是CUDA版本号,我的电脑的CUDA版本号就是11.0
在这里插入图片描述

3.查看python版本

windows键加R键,打开搜索框,输入cmd
在这里插入图片描述输入python,可以看到python版本号为3.6.7
在这里插入图片描述

4.查询适合CUDA版本的torch和torchvision版本

添加链接描述从pytorch官网查询适合CUDA版本的torch和torchvision版本在这里插入图片描述
从官网上查找可知,11.0版本的CUDA,windows系统应该安装pytorch版本为1.7.0,torchvision版本为0.8.0

5.将pytorch和torchvision下载到本地

添加链接描述下载到本地,放到创建的环境地址的文件夹内,不要含有中文路径。
在这里插入图片描述
注意:torch-1.7.0+cu110-cp36-cp36m-win_amd64.whl 中
cu110是指CUDA版本为11.0,cp36是指python版本为3.6
一定不要下错版本,否则会导致安装不成功
除此之外,不建议使用3.6.1版本的python,因为这个版本非常不稳定,会出现很多莫名其妙的问题。
建议使用3.6.7

6.本地安装

.运行cmd,进入环境,输入pip install 环境地址+pytorch安装包名称。
在这里插入图片描述
然后相同方法安装torchvision。

7.验证安装是否成功

cmd窗口输入python,进入python环境后,输入import torch,没有报错,说明安装成功。

在这里插入图片描述
pytorch安装花费了我整整两天时间,在线安装总是失败,下载不是超时,就是连接中断,一两个小时的下载就白费了,这个方法如果中途断网,下一次还能继续下载。torch的安装包比较大,下载时间较长,大家不要心急,祝大家都可以顺利安装。

  • 10
    点赞
  • 39
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值