首页 > 其他分享 >pytorch环境下查看gpu是否可用

pytorch环境下查看gpu是否可用

时间:2022-08-20 13:01:14浏览次数:92  
标签:查看 python torch pytorch cuda gpu

(36条消息) yolov5设置GPU - CSDN

python
#输入库
import torch
#查看版本
print(torch.__version__)
#查看gpu是否可用
torch.cuda.is_available()
#返回设备gpu个数
torch.cuda.device_count()
#退出python
quit()

 

标签:查看,python,torch,pytorch,cuda,gpu
From: https://www.cnblogs.com/yibeimingyue/p/16607547.html

相关文章

  • conda自动安装cpu版本pytorch
    问题描述由于conda自带的源太慢,换成了清华源,结果无论怎么下载都无法使用cuda。检查conda列表发现装的是cpu版本原因网上众说纷纭,例如删掉cpu-only包,但都是瞎猫碰上死耗......
  • docker构建pytorch
    配置拉取镜像(devel版本有nvcc和cudnn)sudodockerpullpytorch/pytorch:1.6.0-cuda10.1-cudnn7-devel构建容器阿立dockerrun-it--gpusall--namelipytorch1.6-......
  • 夜神模拟器 端口查看
    打开软件所在位置类似名称base.vbox,用记事本打开查看hostport<Forwardingname="port0"proto="1"hostip="127.0.0.1"hostport="60001"guestport="18013"/>......
  • 查看用户在某天刷题后第二天还会再来刷题的平均概率
    查看用户在某天刷题后第二天还会再来刷题的平均概率1、数据准备CREATETABLE`question_practice_detail`(`id`intNOTNULL,`device_id`intNOTNULL,`question......
  • git 查看分支 及版本
    git查看分支原文:https://www.cnblogs.com/kingsonfu/p/9840491.html1、查看本地分支gitbranch2、查看本地及远程所有分支gitbranch-a2、查看本地及远程......
  • 【PyTorch学习笔记】1.Tensor 与 Variable
    在PyTorch0.4.0之前,torch.autograd包中存在Variable这种数据类型,主要是用于封装Tensor,进行自动求导。Variable主要包含下面几种属性。 data:被包装的......
  • SQL SERVER 查看处理死锁方法
    第一步:查看死锁表SELECTOBJECT_NAME(RESOURCE_ASSOCIATED_ENTITY_ID)ASTABLENAME,REQUEST_SESSION_IDASPIDFROMSYS.DM_TRAN_LOCKSWHERERESOURCE_TYPE='OBJEC......
  • 查看服务器的硬件信息(产品的型号,序列号,生产厂家等)
      1、查看服务器型号:dmidecode|grep'ProductName'2、查看主板的序列号:dmidecode|grep'SerialNumber'3、查看系统序列号:dmidecode-ssystem-serial-number4......
  • 查看特定表的备注--postgresql
    1. 获取字段名、类型、注释、是否为空SELECT col_description(a.attrelid,a.attnum)ascomment, format_type(a.atttypid,a.atttypmod)astype, a.attnameasname......
  • 如何查看navicat已连接用户的密码
    1:从注册表中查看加密后的密码。1.1:windows键+r,输入regedit 1.2:在注册表中找到\HKEY_CURRENT_USER\SOFTWARE\PremiumSoft\navicat\servers 并找到UserName......