当前位置: 首页 > news >正文

深度学习:Pytorch安装的torch与torchvision的cuda版本冲突问题与解决历程记录

今天不小心将conda环境中的一个pytorch环境中的torch包给搞混了,将其更新了一下,发生了一些问题:

当时运行了一下这个代码: pip install torchvision --upgrade

导致了环境中包的混乱:

只能说欲哭无泪,当时这个 pytorch环境中我是安装的CUDA11.8的版本应该,后来安装了cpu版本的将原先的给覆盖了,导致运行代码时遇到如下报错:

ValueError: Could not find the operator torchvision::nms. Please make sure you have already registered the operator and (if registered from C++) loaded it via torch.ops.load_library.

遇到“Could not find the operator torchvision::nms”这类错误,通常是因为torchtorchvision版本之间的不兼容问题。您提供的信息显示,您的torch版本是2.1.2+cu118。这个问题可能因为torchvision版本不匹配或者安装有问题导致。

这次无意的更新导致了版本的冲突问题!!!

只能重新安装一遍了,先查看显卡配置:

nvidia-smi

 我的CUDA Version:12.1

前往Pytorch官网进行安装:Start Locally | PyTorch

复制这个安装命令:

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

在pycharm的终端处可以进行安装,输入如下指令:

conda activate pytorch

 pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

安装时间比较长,耐心等待。。。。。。。

安装结束后,查看torch与torchvision的cuda版本:

pip show torch torchvision

验证安装

安装完成后,您可以通过运行以下Python代码来验证torchvision是否正确安装,并且是否能够与torch一起使用CUDA:

import torch
import torchvision

print("Torch version:", torch.__version__)
print("Torchvision version:", torchvision.__version__)
print("CUDA available:", torch.cuda.is_available())
if torch.cuda.is_available():
    print("CUDA version:", torch.version.cuda)

我的测试结果:

说明安装成功!!!

 最后测试一下代码运行:

没有报错,又可以继续学习Pytorch了!!! 

相关文章:

  • Django学习全纪录:Django开发环境的搭建
  • uniapp富文本文字长按选中(用于复制,兼容H5、APP、小程序三端)
  • DS:八大排序之归并排序、计数排序
  • PyCharm 取消所有断点
  • 数据模型概念
  • unity学习(20)——客户端与服务器合力完成注册功能(2)调试注册逻辑
  • 计算机专业必看《编程之神》
  • 7.1 Qt 中输入行与按钮
  • 2000-2021年县域指标统计数据库
  • Linux篇:网络基础1
  • 【数据库_MySQL】卸载MySQL
  • Jenkins面试系列
  • 【ArcGIS Pro二次开发】(81):文本符号_CIMTextSymbol
  • (三十五)大数据实战——Superset可视化平台搭建
  • LVS/DR配置
  • 【黑客渗透】-- 远程木马控制
  • 【Effective Objective - C 2.0】——读书笔记(五)
  • Linux:docker在线仓库(docker hub 阿里云)基础操作
  • 微信小程序新手入门教程四:样式设计
  • 多进程服务器和多线程服务器
  • 笔墨如何“构城”?上海美院城市山水晋京展出
  • 招行一季度净利372.86亿降2.08%,营收降逾3%
  • 俄宣布停火三天,外交部:希望各方继续通过对话谈判解决危机
  • 丁俊晖连续7年止步世锦赛16强,中国军团到了接棒的时候
  • 赛力斯拟赴港上市:去年扭亏为盈净利59亿元,三年内实现百万销量目标
  • 朝鲜派兵库尔斯克是否有助于解决乌克兰危机?外交部回应