首页 > 其他分享 >torch._C._cuda_setDevice(device)

torch._C._cuda_setDevice(device)

时间:2023-08-15 21:03:38浏览次数:41  
标签:torch pytorch cuda 版本 ._ device 安装

1.问题原因:安装的事pytorch CPU版本;

torch._C._cuda_setDevice(device)_正常运行

2解决办法:

卸载已安装的pytorch,安装GPU 版本的pytorch

安装命令如下:

pip3 install torch torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple

安装过程有点长,安装过程运行如下:

torch._C._cuda_setDevice(device)_安装过程_02

安装完后验证是否安装成功:

import torch
torch.cuda.is_available()

torch._C._cuda_setDevice(device)_正常运行_03

显示True,代表安装成功啦!

接着就可以正常运行自己的代码啦。。。

标签:torch,pytorch,cuda,版本,._,device,安装
From: https://blog.51cto.com/u_12727662/7092983

相关文章

  • Linux安装Solr-8.9.0
    Solr的工作原理可以简单地概括为以下几个步骤:1.索引创建:首先,Solr需要创建一个索引,用于存储要搜索的数据。索引是基于ApacheLucene构建的,它将文档拆分为字段,并对字段进行分析和标记化,以便进行更有效的搜索和匹配。2.数据导入:Solr可以从多种数据源导入数据,包括数据库、文件、Web......
  • LeetCode 121.买卖股票的最佳时机
    1.题目:给定一个数组 prices ,它的第 i 个元素 prices[i] 表示一支给定股票第 i 天的价格。你只能选择 某一天 买入这只股票,并选择在 未来的某一个不同的日子 卖出该股票。设计一个算法来计算你所能获取的最大利润。返回你可以从这笔交易中获取的最大利润。如果你不能获......
  • AcWing 854. Floyd求最短路
    题目给定一个$n$个点$m$条边的有向图,图中可能存在重边和自环,边权可能为负数。再给定$k$个询问,每个询问包含两个整数$x$和$y$,表示查询从点$x$到点$y$的最短距离,如果路径不存在,则输出impossible。数据保证图中不存在负权回路。输入格式第一行包含三个整数$n,m,k......
  • FL Studio发布21.1新版!新增Hyper Chorus插件及自动更新功能
    很高兴地宣布在去年12月发布重大版本更新后,FLStudio在2023年8月正式更新到21.1版。本次更新虽然只是维护性质,但我们还是为大家带来了一些全新的功能,包括通过钢琴卷中的音阶捕捉和自定义音符工具,引入更快、更有创意的音符编辑。彩色波形,更好地管理采样。极致的合唱插件"HyperChor......
  • Red Hat 8.5下安装19c的GI,11.2.0.4的DB,遇到哪些错误
    需要在RedHat8.5下安装19c的GI,11.2.0.4的DB,安装过程中遇到一些小问题,记录下来。 遇到的错误1:在安装19c的GI,执行./gridSetup.sh时,安装界面报错,如下所示。 [INS-08101]Unexpectederrorwhileexecutingtheactionatstate:'supportedOSCheck' CAUSE-Noadditionali......
  • PyTorch神经网络工具箱-新手笔记
    神经网络核心组件利用PyTorch神经网路工具箱设计神经网络就像搭积木一样,可以极大简化构建模型的任务。神经网络核心组件如下:层:神经网络的基本结构,将输入张量转换为输出张量。模型:由层构成的网络。损失函数:参数学习的目标函数,通过最小化损失函数来学习各种参数。优化器:如在使损失值......
  • vue.js框架的iframe页面计时器无法销毁的解决方法
    同学试过使用生命周期等方式都不能清除计时器;因而改用这个方法;1,首先vue页面上随便写个有高度的div如下:用refs获取高度<divclass="hub-fixed-box":style="{width:fixedWidth+'px'}"ref="fixedTop"></div>2.定时器定义在data内data:{timer:null,//计时器}3,初始......
  • 聊聊JDK1.0到JDK20的那些事儿
    1.前言最近小组在开展读书角活动,我们小组选的是《深入理解JVM虚拟机》,相信这本书对于各位程序猿们都不陌生,我也是之前在学校准备面试期间大致读过一遍,emm时隔多日,对里面的知识也就模糊了。这次开始的时候从前面的JDK发展史和JVM虚拟机家族着手,之前都是粗略读过,这次通过查阅相关......
  • .Net的性能调优
    .Net的性能调优一、.NetCore中的内存泄露1.1什么是内存溢出和内存泄露什么叫内存泄露(memoryleak)是指程序在申请内存后,无法释放已申请的内存空间,导致系统无法及时回收内存并且分配给其他进程使用。通常少次数的内存无法及时回收并不会到程序造成什么影响,但是如果在内存本身就......
  • 我的灵感爆棚-1.以往的
    目录前言前言从高中开始,爱上hiphop;不久之后,爱上了band;善变的人,善变的心。错落的霓虹灯下,几个孤单的游走着本是繁华的城市,匆匆来来又去去啊看似拥挤,嘈杂与喧闹并存在这外壳下,人们的心都没在接近就你看我,我看你,交换过眼神后是麻木的感受为何没有用微笑去对待也是没......