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()