win10上GPU版PyTorch的环境配置

发表于 | 分类于 | 阅读量 次

本文主要解决Win10系统上GPU版PyTorch的环境配置问题。

检查配置

按 Win+X 键打开设备管理器，找到显示适配器，查看自己电脑的显卡配置，如下图所示。

我的电脑是NVIDIA GeForce GTX 960M的显卡。

更新驱动

打开NVIDIA官网更新驱动，选择第一步检查的自己电脑的配置。以我的电脑为例，依次为选择“GeForce”，“GeForce 900M Series (Notebooks)”，”GeForce GTX 960M”，“Windows 10 64-bit”，其他不做改动。点击“Search”后，点击“DOWNLOAD”进行下载。

下载后进行安装，安装过程中安装路径可自定义，选择“只安装图形驱动程序”即可。此过程从略。

用conda安装

利用conda安装GPU版PyTorch的好处是，利用Anaconda包管理器安装cudatoolkit，避免了分别安装CUDA和cuDNN的不便。Anaconda包管理器的安装及基本用法可参见其他教程，此处从略。

打开Anaconda Promt，执行conda create -n pt1 python=3.6 spyder，创建环境pt1。

打开PyTorch官网，根据自己的系统配置进行选择，如下图所示。

上图是我电脑的选择。在刚刚建立的环境pt1中运行“Run this Command”对应的命令

1
conda install pytorch torchvision cudatoolkit=9.2 -c pytorch -c defaults -c numba/label/dev

此过程较为耗时，等待全过程安装完毕。

测试

安装完毕后，在pt1环境中打开Spyder，即，可在开始菜单单击“Spyder(pt1)”。

在Ipython中输入

1
2
3
from __future__ import print_function
import torch as t
print(t.cuda.is_available())

若输出为True，则表明安装配置成功。

参考资料