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

具有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博客

相关推荐
大虫小呓1 分钟前
Python 处理 Excel 数据 pandas 和 openpyxl 哪家强?
python·pandas
聽雨2375 分钟前
03每日简报20250705
人工智能·社交电子·娱乐·传媒·媒体
哪 吒13 分钟前
2025B卷 - 华为OD机试七日集训第5期 - 按算法分类,由易到难,循序渐进,玩转OD(Python/JS/C/C++)
python·算法·华为od·华为od机试·2025b卷
二川bro24 分钟前
飞算智造JavaAI:智能编程革命——AI重构Java开发新范式
java·人工智能·重构
acstdm29 分钟前
DAY 48 CBAM注意力
人工智能·深度学习·机器学习
澪-sl43 分钟前
基于CNN的人脸关键点检测
人工智能·深度学习·神经网络·计算机视觉·cnn·视觉检测·卷积神经网络
羊小猪~~1 小时前
数据库学习笔记(十七)--触发器的使用
数据库·人工智能·后端·sql·深度学习·mysql·考研
摸爬滚打李上进1 小时前
重生学AI第十六集:线性层nn.Linear
人工智能·pytorch·python·神经网络·机器学习
HuashuiMu花水木1 小时前
PyTorch笔记1----------Tensor(张量):基本概念、创建、属性、算数运算
人工智能·pytorch·笔记
lishaoan771 小时前
使用tensorflow的线性回归的例子(四)
人工智能·tensorflow·线性回归