search

OutofAi / cudacanvas

Python Module for PyTorch Tensor Visualisation in CUDA Eliminating CPU Transfer
MIT License
26 stars 0 forks source link

ImportError #1

Closed adarc8 closed 5 months ago

adarc8 commented 5 months ago

this is my torch packages installed

torch                    1.12.1+cu113
torchaudio               0.12.1+cu113
torchvision              0.13.1+cu113

after pip install cudacanvas, i tried to import cudacanvas but got: Traceback (most recent call last): File "/home/adarc/.pycharm_helpers/pydev/_pydevd_bundle/pydevd_exec2.py", line 3, in Exec exec(exp, global_vars, local_vars) File "", line 1, in File "/home/adarc/.pycharm_helpers/pydev/_pydev_bundle/pydev_import_hook.py", line 21, in do_import module = self._system_import(name, *args, *kwargs) File "/home/adarc/.conda/envs/venv_py3_10/lib/python3.10/site-packages/cudacanvas/init.py", line 13, in from .cudaGLStream import CudaGLStreamer File "/home/adarc/.pycharm_helpers/pydev/_pydev_bundle/pydev_import_hook.py", line 21, in do_import module = self._system_import(name, args, **kwargs) ImportError: libcudart.so.12: cannot open shared object file: No such file or directory

OutofAi commented 5 months ago

The wheel packages we published are for 2.1.2+cu118 and 2.1.2+cu121, but it's quite easy to create your own custom wheels for any combination of torch version and cuda version from the source, by forking the source and changing the main.yml file. I generated one for you, I take it you running linux. make sure you uninstall the current package

pip uninstall cudacanvas

and then running this so it uses 1.12.1+cu113

pip install cudacanvas --find-links https://github.com/OutofAi/cudacanvas/wiki/1121_cu113 --force-reinstall --no-cache-dir