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/

pytorch安装教程

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

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

  3. 然后进入python,检测是否安装成功

  4. import torch

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

  6. print(x)

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

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

  9. 返回True即为可用