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

相关推荐
水印云3 分钟前
2025精选5款AI视频转文字工具,高效转录秒变文字!
人工智能·音视频
Enougme7 分钟前
python-虚拟试衣
python
袁庭新1 小时前
大学生为什么一定要重视AI的学习?
人工智能·aigc
酷雷曼VR全景1 小时前
为什么要用VR全景?5个答案告诉你
人工智能·科技·vr·vr全景·酷雷曼
华略创新1 小时前
标准化与定制化的平衡艺术:制造企业如何通过灵活配置释放系统价值
大数据·人工智能·制造·crm·管理系统·erp·企业管理
nightunderblackcat1 小时前
新手向:Python制作贪吃蛇游戏(Pygame)
python·游戏·pygame
wei_shuo1 小时前
用于机器学习的 Podman 简介:简化 MLOps 工作流程
人工智能·机器学习·podman
小马哥编程1 小时前
计算机网络:调制解调器
人工智能·计算机网络·语音识别
flysh052 小时前
pyAutoGUI 模块主要功能介绍-(1)鼠标功能
python·计算机外设·鼠标控制
这张生成的图像能检测吗2 小时前
(论文速读)视觉语言模型评价中具有挑战性的选择题的自动生成
人工智能·计算机视觉·语言模型·视觉语言模型