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

相关推荐
机器之心几秒前
逼近5万亿美元!英伟达GTC深夜爆拉市值,Vera Rubin超级芯片首露面
人工智能·openai
大模型真好玩5 分钟前
LangGraph实战项目:从零手搓DeepResearch(四)——OpenDeepResearch源码解析与本地部署
人工智能·agent·mcp
小白黑科技测评10 分钟前
2025 年视频去水印工具实测:擦擦视频双版本解析一键去字幕与多格式兼容能力
java·人工智能·音视频·智能电视·1024程序员节
IT_陈寒17 分钟前
SpringBoot 3.2新特性实战:这5个隐藏功能让我开发效率提升50%
前端·人工智能·后端
CClaris28 分钟前
机器学习中的数学——矩阵与向量基础
人工智能·机器学习·矩阵
TextIn智能文档云平台28 分钟前
AI在处理扫描版PDF时准确率低,如何提升?
人工智能·pdf
无风听海30 分钟前
神经网络之共现矩阵
人工智能·神经网络·矩阵
想名字好难啊竟然不止我一个38 分钟前
清除 Pip 缓存, 释放磁盘空间
python·缓存·pip
Eiceblue44 分钟前
Python 快速提取扫描件 PDF 中的文本:OCR 实操教程
vscode·python·ocr·1024程序员节
拓端研究室44 分钟前
专题:2025中国汽车行业Data+AI数智化转型与全球化白皮书|附340+份报告PDF、数据仪表盘汇总下载
人工智能