pytorch GPU环境搭建（win10，python3.6，pycharm，cuda9.2，anaconda3.5.2）

cuda下载地址：https://developer.nvidia.com/cuda-downloads

pytorch离线安装包下载地址：https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/

pycharm下载地址：http://www.jetbrains.com/pycharm/

anaconda下载地址：https://www.anaconda.com/download/

1.下载安装python3.6.5，可通过cmd下输入python，在交互环境输入print('hello world')查看是否安装成功。不再赘述.。

2.下载cuda并安装，可通过cmd下输入nvcc -V查看版本信息。以测试是否安装成功。

3.下载安装pycharm。激活方法

4.下载安装anaconda，在命令行输入conda create -n env_name python=3.6.5以创建一个虚拟环境env_name。

在anaconda Prompt中输入activate env_name来激活虚拟环境。

输入conda install D:\Mine\pytorch-0.4.1-py36_cuda92_cudnn7he774522_1.tar.bz2来安装pytorch

输入pip install torchvision来安装torchvision

这时候要安装都已经完成了。下面启动pycharm，来配置使用anaconda的环境吧。

打开pycharm的文件菜单栏，选择默认设置，选择Project interpreter，添加我们刚才新建的环境env_name

现在可以新建一个项目，添加一个测试的python文件

import torch

print(torch.__version__)
print(torch.cuda.is_available())

x=torch.randn(1)
if torch.cuda.is_available():
    device=torch.device("cuda")
    y=torch.ones_like(x,device=device)
    x=x.to(device)
    z=x+y
    print(z)
    print(z.to("cpu",torch.double))

输出：

0.4.1
True
tensor([1.2458], device='cuda:0')
tensor([1.2458], dtype=torch.float64)