本文主要解决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 | from __future__ import print_function |
若输出为True,则表明安装配置成功。