pytorch环境搭建

1、CUDA版本查看

运行命令行，输入

nvidia-msi

查看CUDA Version

2、pytorch下载

• 2.1 pytorch官网链接: https://pytorch.org/.

• 2.2 按照CUDA版本选择
  

选项	含义
PyTorch Build	稳定/预览版
Your OS	系统
Package	下载方式
Language	编程语言
Compute Platform	CUDA版本
Run this Command	下载指令

• 2.3 复制Run this Command的指令，在命令行运行
  

• 注意事项
  • 可能是用了校园网的原因，没有遇到下载慢的情况
  • 没有CUDA11.2，选择了CUDA11.3，回头看看有没有雷

• 2.4 验证安装

  运行命令行，输入

  python
  import torch
  torch.cuda.is_available()