tensorflow可以使用GPU,而torch无法使用GPU

embedded/2024/10/11 13:25:37/

具有conda测试环境无意使用pip更新了tensorflow后,导致总报提示“could not load dynamic library ‘libnvinfer.so.7’”,经过查询后建议pip uninstall tensorflow后使用conda安装。

conda install tensorflow后解决了库动态加载问题,但是torch出现无法访问GPU的问题。

卸载torch而后重装,可以恢复GPU。

1)查看cuda的版本:cuda list cuda

2)从pytorch官网查找安装命令:

# CUDA 10.1
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch

 3)验证torch能否访问GPU:

        import torch

        print (torch.cuda.is_available())

参考链接:

1、pytorch 安装命令:Previous PyTorch Versions | PyTorch

2、Could not load dynamic library 'libnvinfer.so.7' 解决方法,Could not load dynamic library 'libnvinfer.so.7' 解决方法-CSDN博客


http://www.ppmy.cn/embedded/33563.html

相关文章

【SQL基础】mysql中如何将日期时间类型转换为日期类型

在MySQL中,将DATETIME类型的数据转换为日期格式可以通过使用DATE()函数来实现。DATE()函数可以从DATETIME或TIMESTAMP类型的值中提取出日期部分。 以下是几种将DATETIME转换为日期格式的示例: 直接转换DATETIME列为日期: SELECT DATE(date…

C#面:简述 C# 种索引器实现过程,是否只能根据数字索引

C# 中的索引器(Indexer)是一种特殊的属性,它允许对象像数组一样通过索引来访问其元素。索引器可以用于访问类、结构体或接口中的元素,使得对象可以像数组一样进行索引操作。 可以通过任意类型进行索引。 索引器的实现过程如下&am…

Android C++ 开发调试 LLDB 工具的使用

文章目录 调试环境准备基础命令Breakpoint CommandsWatchpoint CommandsExamining VariablesEvaluating ExpressionsExamining Thread StateExecutable and Shared Library Query Commands 参考: Android 中在进行 NDK 开发的时候,我们经常需要进行 C 代…

openjudge_2.5基本算法之搜索_2152:Pots

题目 2152:Pots 总时间限制: 1000ms 内存限制: 65536kB 描述 You are given two pots, having the volume of A and B liters respectively. The following operations can be performed: FILL(i) fill the pot i (1 ≤ i ≤ 2) from the tap; DROP(i) empty the pot i to the…

QT:QT与操作系统

文章目录 信号槽与事件QT多线程概述原理完成倒计时程序 UDP回显服务器服务端客户端 信号槽与事件 在之前的信号槽中,已经有了一个基本的认识,那么对于QT中事件的理解其实就非常的类似,当用户进行某种操作的时候,就会触发事件&…

MySQL45讲(一)(43)

这一小节是有关分区表的事情 分区表在mysql8.0的时候已经不允许myisam进行分区表的创建,应为该方法使用的是通用分区策略,然后就是open_file_limits的默认值是1024,然后就要把所有的分区表都要访问个遍,如果分区多一点&#xff0c…

k8s调度原理以及自定义调度器

kube-scheduler 是 kubernetes 的核心组件之一,主要负责整个集群资源的调度功能,根据特定的调度算法和策略,将 Pod 调度到最优的工作节点上面去,从而更加合理、更加充分的利用集群的资源,这也是我们选择使用 kubernete…

pgvector扩展在IvorySQL Oracle兼容模式下的应用实践

向量数据库是生成式人工智能(GenAI)的关键组成部分。作为PostgreSQL的重要扩展,pgvector支持高达16000维的向量计算能力,使得PostgreSQL能够直接转化为高效的向量数据库。 IvorySQL基于PostgreSQL开发,因此它同样支持添加pgvector扩展。在Ora…