安装准备
请先安装Anaconda或者Miniconda并配置好清华源
安装教程
- 用conda新建一个名为
pytroch
的环境
conda create -n pytorch python=3.7
- 激活刚才创建的名为
pytorch
的环境
conda activate pytorch
- 进入
Pytorch
的官网,然后复制下方的安装命令
- 在第2步激活的
pytorch
环境中运行第3步复制的安装命令
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
验证安装
- 验证
torch
能不能正常使用,正常打印则说明安装成功
import torch
x = torch.rand(5, 3)
print(x)
- 再来看看CUDA是否能用,如果打印
True
则说明能用
import torch
print(torch.cuda.is_available())
引用参考
https://pytorch.org/get-started/locally/