增城新塘网站建设,去除tag wordpress,互联网挣钱好项目,logo设计免费生成器在线今天不小心将conda环境中的一个pytorch环境中的torch包给搞混了#xff0c;将其更新了一下#xff0c;发生了一些问题#xff1a;
当时运行了一下这个代码#xff1a; pip install torchvision --upgrade
导致了环境中包的混乱#xff1a; 只能说欲哭无泪#xff0c;当…今天不小心将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”这类错误通常是因为torch和torchvision版本之间的不兼容问题。您提供的信息显示您的torch版本是2.1.2cu118。这个问题可能因为torchvision版本不匹配或者安装有问题导致。 这次无意的更新导致了版本的冲突问题 只能重新安装一遍了先查看显卡配置 nvidia-smi 我的CUDA Version12.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 torchvisionprint(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了
