pytorch安装教程
tensorflow安装教程
在上面那篇博客里面安装过cuda和cudnn了,所以说这里不再写了。有了前面的铺垫pytorch简单很多
我这里假设你anaconda,cuda这些都已经安装好了,然后
- 创建一个pytorch专用环境:conda create -n pytorch python=3.6,其他python版本应该也行
- 启动环境:conda activate pytorch
- 进pytorch官网,寻找合适的下载指令。https://pytorch.org/get-started/locally/
-
指令为:conda install pytorch torchvision cudatoolkit=10.1 -c pytorch但是我们的cuda为10.0版本。所以说我输入的指令为conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
-
等待安装即可,下载有点慢,不行的话删除环境再多试几次
-
然后进入python,检测是否安装成功
-
import torch
-
x=torch.rand(4,3)
-
print(x)
-
成功输出一个四行三列的矩阵即可
-
检测gpu加速是否可用: print(torch.cuda.is_available())
-
返回True即为可用