pytorch安装教程

tensorflow安装教程

在上面那篇博客里面安装过cuda和cudnn了，所以说这里不再写了。有了前面的铺垫pytorch简单很多

我这里假设你anaconda，cuda这些都已经安装好了，然后

1. 创建一个pytorch专用环境：conda create -n pytorch python=3.6，其他python版本应该也行
2. 启动环境：conda activate pytorch
3. 进pytorch官网，寻找合适的下载指令。https://pytorch.org/get-started/locally/

5. 指令为：conda install pytorch torchvision cudatoolkit=10.1 -c pytorch但是我们的cuda为10.0版本。所以说我输入的指令为conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

6. 等待安装即可，下载有点慢，不行的话删除环境再多试几次

7. 然后进入python，检测是否安装成功

8. import torch

9. x=torch.rand(4,3)

10. print(x)

11. 成功输出一个四行三列的矩阵即可

12. 检测gpu加速是否可用： print(torch.cuda.is_available())

13. 返回True即为可用