[PYTHON] Le module PyTorch indique que libcusparse.so.10 est manquant

Notes personnelles

OS : Ubuntu 18.04 LTS PyTorch : 1.5.0, 1.6.0 CUDA : 10.2 NVIDIA-driver : 440

À l'origine, une erreur survenue lors de l'importation de PyTorch Geometric. J'ai trouvé étrange que je l'ai installé officiellement ... mais en conclusion, le pilote NVIDIA a été installé correctement, mais NVCC n'a pas été installé.

Cas

import torch_geometric.transforms as T

OSError: libcusparse.so.10: cannot open shared object file: No such file or directory

ImportError de libcublas.so.10.0 dans tensorflow-gpu https://qiita.com/Uejun/items/fbb579374eafab8633d6

C'est un cas similaire, mais il semble que vous n'ayez qu'à le réussir. Cependant, peu importe où je regardais, mon ordinateur n'avait pas de bibliothèque appelée libcusparse.so.10. Il y avait quelque chose comme ça dans le numéro officiel.

libcusparse.so.10 error when importing https://github.com/rusty1s/pytorch_geometric/issues/1092

La situation du problème est différente, mais il semble que libcusparse.so.10 se trouve sous / usr / local / cuda / lib64 et que $ nvcc -V doit passer correctement. Dans mon cas, $ nvcc -V n'est pas passé en premier lieu. Lorsque vous appuyez sur cette commande

$ nvcc -V

Command 'nvcc' not found, but can be installed with:

sudo apt install nvidia-cuda-toolkit

Est affiché, mais $ sudo apt install nvidia-cuda-toolkit Dans , la bibliothèque cible n'était pas incluse. Avant cela, il semble que cuda 10.2 doit être installé.

Solution

Réinsérez cuda 10.2.

$  sudo apt purge nvidia-cuda-* 

Après (juste au cas où vous n'auriez pas à le faire), basé sur Officiel Installez cuda 10.2. Je vais omettre cela car je pense que c'est différent pour chaque appareil. Après cela, dans .bashrc

export PATH="/usr/local/cuda/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda/lib64:$LD_LIBRARY_PATH"

Après l'écriture, $ nvcc -V est passé et PyTorch Geometric est devenu disponible. De plus, s'il existe un module tel que libucusparse.so.10.0, il semble qu'il soit possible de traiter $ ln -s libucusparse.so.10.0 libucusparse.so.10, mais dans mon cas, une erreur s'est produite.

Recommended Posts

Le module PyTorch indique que libcusparse.so.10 est manquant
[Tutoriel PyTorch ①] Qu'est-ce que PyTorch?
PyTorch DataLoader est lent