windows10+python3.5+CUDA10.0 安装pytorch教程

安装环境准备:

1.首先你需要安装Anaconda:

https://www.anaconda.com/

需要注意Anaconda自带的python版本。我安装的Anaconda自带的是python3.5版本。

2.其次你需要安装CUDA 10.0和对应的cuDNN v7(见相应的博客教程)。

安装pytorch

官网(https://pytorch.org/get-started/locally/)给出的安装命令是:

CUDA 10.0

# Python 3.5
pip3 install https://download.pytorch.org/whl/cu100/torch-1.0.1-cp35-cp35m-win_amd64.whl
pip3 install torchvision

上述两个命令对应的包都需要安装。实测发现由于安装包很大(698M) ,直接用如上命令安装下载时频频出错,无法安装。

正确安装方法如下:

1.离线安装上述whl文件。你可以直接将intall后面的网址输入浏览器下载,但实测下载成功率也不高。所以我将此文件上传至百度网盘:

链接:https://pan.baidu.com/s/1i3E8Na08UKSajRMKLCBrMg 
提取码:amk4 
下载后使用命令

pip install (whl文件的存放路径)/torch-1.0.1-cp35-cp35m-win_amd64.whl

即可安装成功。

 2.实测第二条安装命令(pip3 install torchvision)也无法安装成功,显示找不到安装包。解决方法:

先更新pip:

python -m pip install --upgrade pip

随后使用清华源安装:

pip3 install torchvision -i https://pypi.tuna.tsinghua.edu.cn/simple

至此pytorch就已经安装成功啦!在命令行输入如下命令测试一下:

python
from __future__ import print_function
import torch
x = torch.rand(5, 3)
print(x)

 此命令生成一个满足0-1均匀分布的矩阵:

tensor([[0.3380, 0.3845, 0.3217],
        [0.8337, 0.9050, 0.2650],
        [0.2979, 0.7141, 0.9069],
        [0.1449, 0.1132, 0.1375],
        [0.4675, 0.3947, 0.1426]])

 测试一下CUDA是否可用:

import torch      #之前import过了此句可不写
torch.cuda.is_available()

显示true就表示没有问题。

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值