安装准备

  1. 请先安装Anaconda或者Miniconda并配置好清华源
  2. 输入命令nvidia-smi,查看CUDA版本,可以看到是10.1
    在这里插入图片描述

安装教程

  1. 用conda新建一个名为pytroch的环境
conda create -n pytorch python=3.7
  1. 激活刚才创建的名为pytorch的环境
conda activate pytorch
  1. 进入Pytorch官网CUDA版本选择之前我们查询的。然后复制下方的安装命令
    在这里插入图片描述
  2. 在第2步激活的pytorch环境中运行第3步复制的安装命令
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

验证安装

  1. 验证torch能不能正常使用,正常打印则说明安装成功
import torch
x = torch.rand(5, 3)
print(x)
  1. 再来看看CUDA是否能用,如果打印True则说明能用
import torch
print(torch.cuda.is_available())

引用参考

https://pytorch.org/get-started/locally/
Logo

更多推荐