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 分钟前
LangChain加载HTML内容全攻略:从入门到精通
python·langchain
静心问道3 分钟前
TrOCR: 基于Transformer的光学字符识别方法,使用预训练模型
人工智能·深度学习·transformer·多模态
说私域5 分钟前
基于开源AI大模型、AI智能名片与S2B2C商城小程序源码的用户价值引导与核心用户沉淀策略研究
人工智能·开源
亲持红叶6 分钟前
GLU 变种:ReGLU 、 GEGLU 、 SwiGLU
人工智能·深度学习·神经网络·激活函数
说私域7 分钟前
线上协同办公时代:以开源AI大模型等工具培养网感,拥抱职业变革
人工智能·开源
群联云防护小杜8 分钟前
深度隐匿源IP:高防+群联AI云防护防绕过实战
运维·服务器·前端·网络·人工智能·网络协议·tcp/ip
惜.己12 分钟前
使用python读取json数据,简单的处理成元组数组
开发语言·python·测试工具·json
摘星编程14 分钟前
构建智能客服Agent:从需求分析到生产部署
人工智能·需求分析·智能客服·agent开发·生产部署
不爱学习的YY酱17 分钟前
信息检索革命:Perplexica+cpolar打造你的专属智能搜索中枢
人工智能
whaosoft-1431 小时前
51c自动驾驶~合集7
人工智能