cuda9.1 Win10 python3.6版本的pytorch环境安装

由于大创需要机器学习的框架pytorch,安装过程中这里写一下下载配置过程中遇到的坑。

cpu版本:

只需要复制官网中自己对应版本的pip下载指令就可以了,网速还是很快的

在这里插入图片描述
然而下载后才发现代码是需要gpu的

在控制面板中查看自己的cuda版本

(1)按win+Q,输入控制面板,然后查看方式小图标,点击NVIDIA控制面板:

img

点击NVIDIA控制面板的帮助,点击左下角系统信息:

img

点击组件:这里就显示了你的CUDA的信息啦!!!

img

[图片源自 https://blog.csdn.net/qq_38295511/article/details/89223169 ]

我的CUDA版本是 9.1.84,python版本3.6.2 win10

使用如下语句可以安装cu91版本的pytorch

pip3 install http://download.pytorch.org/whl/cu91/torch-0.4.0-cp36-cp36m-win_amd64.whl 

然而下载torchvision会使用之前缓存中的torchvision而且0.4.1版本的并不能使用

最后使用

pip install torchvision==0.1.8 

成功安装

  • 7
    点赞
  • 25
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值