使用海关产品(8\*K100Ai)的过程中碰到了一个问题:
docker pull harbor.sourcefind.cn:5443/dcu/admin/base/pytorch:2.5.1-ubuntu22.04-dtk25.04.2-py3.10
2.5.1-ubuntu22.04-dtk25.04.2-py3.10: Pulling from dcu/admin/base/pytorch
拉取了海光dtk的容器,用来运行算法,一开始提示我:
Traceback (most recent call last):
File "/audio\_algorithms/das\_algorithm.py", line 3, in
```
import librosa
```
ModuleNotFoundError: No module named 'librosa'
于是我pip install了这个librosa之后,继续运行,但是报错:
root@MiWiFi-RD05-srv:/audio\_algorithms# python das\_algorithm.py
Traceback (most recent call last):
File "audio\_algorithms/das\_algorithm.py", line 4, in
```
import torchaudio
```
File "/usr/local/lib/python3.10/dist-packages/torchaudio/\_\_init\_\_.py", line 2, in
```
from . import \_extension # noqa # usort: skip
```
File "/usr/local/lib/python3.10/dist-packages/torchaudio/\_extension/\_\_init\_\_.py", line 38, in
```
\_load\_lib("libtorchaudio")
```
File "/usr/local/lib/python3.10/dist-packages/torchaudio/\_extension/utils.py", line 60, in \_load\_lib
```
torch.ops.load\_library(path)
```
File "/usr/local/lib/python3.10/dist-packages/torch/\_ops.py", line 1350, in load\_library
```
ctypes.CDLL(path)
```
File "/usr/lib/python3.10/ctypes/\_\_init\_\_.py", line 374, in \_\_init\_\_
```
self.\_handle = \_dlopen(self.\_name, mode)
```
OSError: libtorch\_cuda.so: cannot open shared object file: No such file or directory
根据我的估计,这个问题应该是由于直接pip安装了cuda的相关库导致,有没有什么办法解决呢?或者有没有完整支持pytorch+torchaudio+librosa的镜像存在呢?