win10+conda+pytorch-gpu安装

pytorch 官网上只有linux和Mac的程序包,没有windows系统的,但是windows系统还是可以用pytorch的。 
github: https://github.com/peterjc123/pytorch-scripts

首先要安装anaconda:

推荐清华镜像 
https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/ 
拖到最后,下载最新版即可。

这里写图片描述

安装完毕要安装pytorch运行环境:

# If your main Python version is not 3.5 or 3.6
conda create -n pytorch python=3.6 numpy pyyaml mkl
  • 3. activate pytorch36 激活这个虚拟环境(取消激活用deactivate)

    4. 在pytorch 官网上寻找自己的环境,注意自己电脑是不是有显卡,查看自己是否是独显还是集成显卡(集成显卡就是核显)在这里不能使用cuda 安装,按win+R 输入dxdiag 找到显示,查看显卡属性,然后百度搜是否是集成显卡还是独显

    显然我的电脑是有独立的显卡,在这里我用的是pip 安装,

  • 输入

  • pip3 install http://download.pytorch.org/whl/cu80/torch-0.4.0-cp36-cp36m-win_amd64.whl 
    pip3 install torchvision
  • 还有一种是conda 安装

然后是安装gpu版本,pytorch:

# for Windows 10 and Windows Server 2016, CUDA 8
conda install -c peterjc123 pytorch cuda80

因为我的TensorFlow用 的CUDA8,所以我安装的是 cuda8,你也可已安装新版本试试

这里写图片描述

安装Visual C++ 2015 64 bit

https://www.microsoft.com/zh-cn/download/details.aspx?id=48145 
有就不用装了,一般玩游戏都有的。

下载安装Cuda Toolkit 8.0

https://developer.nvidia.com/cuda-downloads

因为TensorFlow只能用 CUDA8,我装的8.0,不过听说pytorch支持9.0,可以尝试。

这里写图片描述

下载8.0:

这里写图片描述

选择版本:

这里写图片描述

两个都要下载:

这里写图片描述

下载完毕安装就行了。

安装cuDNN

官网维护了,度盘链接在此: 
https://pan.baidu.com/s/1nwAYgIl

解压后覆盖到Cuda Toolkit 8.0安装目录中,这样就可以使用了。

测试一下: 
打开python,输入如下代码:

import torch
torch.cuda.is_available()
  •  

返回True证明成功了。

参考资料:

https://blog.csdn.net/tonydz0523/article/details/79020626

https://blog.csdn.net/u014439564/article/details/80270996

猜你喜欢

转载自blog.csdn.net/barry_j/article/details/81079218