Pytorch学习笔记——Pytorch基础

安装过程中因为网速不好，需要把channel换到了清华的源，可以参考知乎大神的文章。

安装中出现 无法将“conda”项识别为 cmdlet、函数、脚本文件或可运行程序的名称。请检查名称的拼写，如果包括路径，请确保路径正确，然后再试一次的问题，
管理员身份运行CMD，并在conda命令前加上“./”(Win10)安装顺利完成。
使用以下代码测试，同时测试安装的CUDA和CUDNN是否可用。

# CUDA TEST
import torch
a = torch.Tensor([1.0])
cu = a.cuda()
print(cu)

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

如果cuDNN不可用，可以将其关闭；

cudnn.enabled = False

reference
https://zhuanlan.zhihu.com/p/26871672