ubuntu python 查看系统是否可用 cuda

  • 初
  • 环境与设备
  • 查看环境
  • 演示效果


本篇文章将介绍ubuntu python 查看系统是否可用 cuda

初

希望能写一些简单的教程和案例分享给需要的人

环境与设备

系统：ubuntu 22.04
工具：python 3.10.6
设备：Nvidia GeForce RTX 4090

查看环境

import torch
print(torch.__version__)
print(torch.cuda.is_available())

第一个打印输出：

打印出当前导入的 PyTorch 版本。torch.version 是 PyTorch 模块中的一个变量，它包含了当前安装的 PyTorch 版本号。

第二个打印输出：

打印出一个bool值，查看系统是否可用 NVIDIA 的 CUDA 技术进行 GPU 加速计算。如果输出为 True，表示你的系统支持 CUDA 加速；如果输出为 False，则表示不支持。

演示效果

先命令进入python3

python3

ubuntu python 查看系统是否可用 cuda_CUDA

然后我们可以看到系统输出了：

Python 3.10.6 (main, Nov 14 2022, 16:10:14) [GCC 11.3.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>>

我们输入：

import torch

返回提示：

Python 3.10.6 (main, Nov 14 2022, 16:10:14) [GCC 11.3.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ModuleNotFoundError: No module named 'torch'
>>>

说明我们，没有安装 torch 模块，我们去官网查找安装的方式

进入下面网址：

https://pytorch.org/

进入后往下滚动查看到一个可以选系统选版本的地方：

ubuntu python 查看系统是否可用 cuda_linux_02

根据自己的需求来选择

我选择了预览版，采用了 Linux 平台上的 pip 安装方式，同时需要使用 CUDA 12.1 版本。选择完成后，在“Run this Command” 部分找到命令，复制并粘贴到我们的命令控制台执行。

pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121

ubuntu python 查看系统是否可用 cuda_CUDA_03


安装完成后，回到python3这边

一行行输入

import torch
print(torch.__version__)
print(torch.cuda.is_available())

就可以看到效果：

ubuntu python 查看系统是否可用 cuda_ubuntu_04


阿里云国内75折 回扣 微信号：monov8
阿里云国际，腾讯云国际，低至75折。AWS 93折 免费开户实名账号 代冲值 优惠多多 微信号：monov8 飞机：@monov6
标签: pythonUbuntu