首页 » Windows » windows10 安装 pytorch gpu版本

windows10 安装 pytorch gpu版本

原文 http://blog.csdn.net/tonydz0523/article/details/79020626

2018-01-11 02:01:58阅读(473)

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

首先要安装anaconda:

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

<a href=windows10 安装 pytorch gpu版本" src="http://img.blog.csdn.net/20180110103334227?watermark/2/text/aHR0cDovL2Jsb2cuY3Nkbi5uZXQvdG9ueWR6MDUyMw==/font/5a6L5L2T/fontsize/400/fill/I0JBQkFCMA==/dissolve/70/gravity/SouthEast" alt="这里写图片描述" title="">

安装完毕要安装pytorch运行环境:
# If your main Python version is not 3.5 or 3.6
conda create -n test python=3.6 numpy pyyaml mkl

然后是安装gpu版本,pytorch:

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

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

windows10 安装 pytorch  gpu版本

安装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,可以尝试。

windows10 安装 pytorch  gpu版本

下载8.0:

windows10 安装 pytorch  gpu版本

选择版本:

windows10 安装 pytorch  gpu版本

两个都要下载:

windows10 安装 pytorch  gpu版本

下载完毕安装就行了。

安装cuDNN

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

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

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

import torch
torch.cuda.is_available()

返回True证明成功了。

windows10 安装 pytorch  gpu版本

结束。。。。。。。。。。。。。。。。。。。。。。

最新发布

CentOS专题

关于本站

5ibc.net旗下博客站精品博文小部分原创、大部分从互联网收集整理。尊重作者版权、传播精品博文,让更多编程爱好者知晓!

小提示

按 Ctrl+D 键,
把本文加入收藏夹