记64位Win10安装Pytorch过程

1、安装anaconda3:

https://www.anaconda.com/download/#windows

选Python3.6的x64版本下载、安装。

安装之后,把“C:\ProgramData\Anaconda3;C:\ProgramData\Anaconda3\Library\bin”加到系统环境变量(可选,主要为了后面运行conda命令)。

 

2、安装 CUDA。注意:需要有Nv显卡,否则没必要装了。目前最新的9.1版本:

https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10

下载network或local包都可以安装 。

 

3、下载 安装anaconda下的Pytorch包:

参考 蒲嘉宸 大神的文章:https://zhuanlan.zhihu.com/p/26871672

下载地址:https://pan.baidu.com/s/1dF6ayLr#list/path=%2F

根据CUDA是7、8、9选择 带有cu70、cu80、cu90的最新安装包下载。我选的是pytorch-0.3.1-py36_cuda90_cudnn7he774522_2.tar.bz2

下载完之后,运行:

cd "下载包的路径"
conda install numpy mkl cffi
conda install --offline pytorch????.tar.bz2

 

完成。可以运行代码进行测试。

 

posted on 2018-03-06 00:36  东篱南山  阅读(11769)  评论(1编辑  收藏  举报

导航