Pytorch rocm 安装错误。如何解决这个问题?

问题描述 投票:0回答:0

我已经使用以下命令以 pytorch.org 上推荐的方式安装了 pytorch:

sudo pip3 install torch torchvision --extra-index-url https://download.pytorch.org/whl/rocm5.2

但是安装后,我用pytorch运行一个python脚本并得到错误:

/home/vic/.local/lib/python3.10/site-packages/torchvision/io/image.py:13: UserWarning: Failed to load image Python extension: libc10_cuda.so: cannot open shared object file: No such file or directory
  warn(f"Failed to load image Python extension: {e}")

如何解决?

手电筒版本:

import torch
torch.__version__

'1.13.1+rocm5.2'

torchvision版本(错误):

import torchvision
/home/vic/.local/lib/python3.10/site-packages/torchvision/io/image.py:13: UserWarning: Failed to load image Python extension: libc10_cuda.so: cannot open shared object file: No such file or directory
  warn(f"Failed to load image Python extension: {e}")
pytorch amd-rocm
© www.soinside.com 2019 - 2024. All rights reserved.