(机器学习)Windows系统下的Pytorch配置

前面一些软件下载和基础搭建工作可见另一篇博文<(机器学习)Windows系统下的Tensorflow-gpu配置>

Pytorch配置

1. 找到新安装的Anaconda中的Prompt，管理员身份打开：
   

2. 创建一个虚拟环境（在此环境内可以任意配置而不会影响系统的环境）

   conda create -n 环境名 -c 镜像源（python版本）

   用户安装的不同python环境都会被放在目录~/anaconda/envs下，可以在命令中运行conda info -e查看已安装的环境，当前被激活的环境会显示有一个星号或者括号。

   删除虚拟环境:

   使用命令conda remove -n 虚拟环境名称 --all， 即可删除。

   删除环境中的某个包：

   使用命令conda remove --name $虚拟环境名称 $package_name 即可

3. 激活或关闭环境

   安装完后激活环境：

   activate py36

   关闭环境：

   deactivate

4. 安装pytorch所需要的包

conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

注：-c pytorch是强制从官网下载，如果下载不了可以删去
5. 测试

# CUDA TEST
import torch
x = torch.Tensor([1.0])
xx = x.cuda()
print(xx)

# CUDNN TEST
from torch.backends import cudnn
print(cudnn.is_acceptable(xx))