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就表示没有问题。

你可能感兴趣的:(深度学习,pytorch)