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)