windows10 pytorch 安装

Windows10 pytorch 安装

1.查看电脑cuda版本决定安装的Pytorch版本

windows10 pytorch 安装
help=>system infomation
windows10 pytorch 安装
以我电脑为例,cuda8 python 版本为3.6

2.下载pytorch

尝试了很多方法 即使使用pip install -i 清华镜像 依旧很慢
最佳方法:
直接进入清华大学开源软件镜像站https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
下载相对应的版本 到本地 事实证明这种方法下载的很快

3.安装

打开 Anaconda prompt, acrtivate 你的python 环境, 接着进入到上一步下载的路径下,使用命令:
conda install --offline pytorch-0.4.1 py36_cuda80_cudnn7he774522_1.tar.bz2

–offline后面是你下载的文件名称

成功