安装Pytorch环境（cuda+Pytorch）参考资料： 60 分钟快速入门 PyTorch

来自 🥬🐶 程序员 Truraly | 田园 的博客

文章首发于：田园幻想乡 | PyTorch

Github 交友：Truraly

参考资料：

60 分钟快速入门 PyTorch | 知乎

安装Pytorch环境（cuda+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))

————————————————