今天,看一篇paper中用的是pytorch实现,所以安装一下这个框架,以便以后学习,由于之前安装过tensorflow,所以此次安装pytorch很顺利,省去了不少麻烦。
在win10 python3.6开启Pytorch支持:
你需要满足以下条件:
- Anaconda3 (with Python 3.5/3.6)
- Windows 64位系统(Windows 7 或 Windows Server 2008 及以上)
- CUDA 8.0.61
- cuDNN v5以上
conda install -c peterjc123 pytorch_legacy
(我是科学上网的 ,下载速度还可以)
由于我在之前安装了tensorflow,所以省去了安装cuda,cudnn.
简单测试CUDA与cuDNN是否工作正常的代码:
# 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))
哈哈,大功告成!