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

相关推荐
张较瘦_28 分钟前
[论文阅读] 人工智能 + 软件工程 | 需求获取访谈中LLM生成跟进问题研究:来龙去脉与创新突破
论文阅读·人工智能
一 铭1 小时前
AI领域新趋势:从提示(Prompt)工程到上下文(Context)工程
人工智能·语言模型·大模型·llm·prompt
云泽野4 小时前
【Java|集合类】list遍历的6种方式
java·python·list
麻雀无能为力5 小时前
CAU数据挖掘实验 表分析数据插件
人工智能·数据挖掘·中国农业大学
时序之心5 小时前
时空数据挖掘五大革新方向详解篇!
人工智能·数据挖掘·论文·时间序列
IMPYLH5 小时前
Python 的内置函数 reversed
笔记·python
.30-06Springfield5 小时前
人工智能概念之七:集成学习思想(Bagging、Boosting、Stacking)
人工智能·算法·机器学习·集成学习
说私域6 小时前
基于开源AI智能名片链动2+1模式S2B2C商城小程序的超级文化符号构建路径研究
人工智能·小程序·开源
永洪科技6 小时前
永洪科技荣获商业智能品牌影响力奖,全力打造”AI+决策”引擎
大数据·人工智能·科技·数据分析·数据可视化·bi
shangyingying_17 小时前
关于小波降噪、小波增强、小波去雾的原理区分
人工智能·深度学习·计算机视觉