Windows上的Pytorch安装

Windows上Pytorch安装有3个方案:

  1. Pytorch官网安装
     (-c pytorch)直接https://pytorch.org/(Pytorch官网,按需求选择cuda,py版本,按正常人思维选择会自动给出代码,复制执行即可),其中本地cuda版本查看方式:桌面右键NVIDIA控制面板->左下角系统信息->组件->NVCUDA.DLL右边就是了
  2. 上面基本不行,国内通过清华源下载,配置源后去除-c pytorch,我试了也不行
  3. 本地下载whl文件:试了可行,通过第1步基本就知道自己要下上面版本的torch和torchvision,两个要搭配用的,都要知道自己需要的版本

第3个方法的具体步骤

  • 点击有torch和torchvision版本的选择下载 ,选择下载需要的
  • 打开cmd或者conda环境窗口,pip install D:\文件名.whl自动下载
  • 在conda list或者pip list里能看到下载成功的库
    我的torch和torchvision通过第1步观察到需要的两个包名是:
    torch-1.2.0+cu92-cp37-cp37m-win_amd64.whl
    torchvision-0.4.0+cu92-cp37-cp37m-win_amd64.whl
    因为要和cuda对应以及两个包要搭配使用,费了很多劲,记录下方便查看。

pass1:如果下载库时候很慢甚至停止,再次打开会飞快~
pass2:记得我下载完运行时候显示PILLOW库版本不对,用pip install PILLOW=xxx降了版本的

最关键的链接给出来:https://download.pytorch.org/whl/torch_stable.html

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值