安装Pytorch环境（cuda+Pytorch）

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

• 文章首发于：田园幻想乡 | PyTorch
• Github 交友：Truraly

---

参考资料：

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

安装Pytorch环境（cuda+Pytorch）

参考链接：

• Linux 查看当前 Cuda（CUDA Toolkit ）版本 | 优快云
• 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))

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

版权声明：本文为 田园幻想乡 的原创文章 ，并亲自发布于 优快云 平台 。遵循 CC 4.0 BY-NA-SA 版权协议，转载请附上原文出处链接及本声明。
原文链接：http://truraly.fun/学习笔记/数据科学/PyTorch.html