PyTorch

目录

• 安装Pytorch环境（cuda+Pytorch）

---

参考资料：

• 60 分钟快速入门 PyTorch | 知乎

安装Pytorch环境（cuda+Pytorch）

参考链接：

• Linux 查看当前 Cuda（CUDA Toolkit ）版本 | CSDN
• Ubuntu22.04+Pytorch 环境配置全过程 | 知乎

---

首先需要装 nv 的显卡驱动

使用nvidia-smi可以看到显卡的版本能装的 cuda 的最高版本

然后安装 cuda，不同版本 cuda 对显卡驱动的最低要求不同

cuda 下载页面

适用于 Linux 的 NVIDIA CUDA 安装指南 | 微软官方

安装完 cuda 之后

使用nvcc -V或者cat /usr/local/<cuda_path>/version.txt 查看 cuda 版本

torch 下载页面 选择合适的版本下载

PyTorch 历史版本

使用 python -c "import torch; print(torch.cuda.is_available())"查看是否可以使用 GPU

使用如下代码获取更详细的信息

import sys
import torch

print("Python version")
print(sys.version)

print("\nPyTorch version")
print(torch.__version__)

print("\nIs CUDA available for PyTorch")
print(torch.cuda.is_available())

if torch.cuda.is_available():
    print("\nCUDA version")
    print(torch.version.cuda)

    print("\nGPU details")
    print(torch.cuda.get_device_name(0))