网站开发费用摊销时间,html5视频教程,wordpress在线扫描,百度数据研究中心官网ubuntu python 查看系统是否可用 cuda 初环境与设备查看环境演示效果 本篇文章将介绍ubuntu python 查看系统是否可用 cuda
初
希望能写一些简单的教程和案例分享给需要的人
环境与设备
系统#xff1a;ubuntu 22.04 工具#xff1a;python 3.10.6 设备#xff1a;Nvidi… 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然后我们可以看到系统输出了
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/进入后往下滚动查看到一个可以选系统选版本的地方 根据自己的需求来选择
我选择了预览版采用了 Linux 平台上的 pip 安装方式同时需要使用 CUDA 12.1 版本。选择完成后在“Run this Command” 部分找到命令复制并粘贴到我们的命令控制台执行。
pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121 安装完成后回到python3这边
一行行输入 import torch
print(torch.__version__)
print(torch.cuda.is_available())就可以看到效果
