运行torch报错cuda

Question

OSError: /home/anaconda3/lib/python3.7/site-packages/torch/lib/../../nvidia/cublas/lib/libcublas.so.11: symbol cublasLtHSHMatmulAlgoInit version libcublasLt.so.11 not defined in file libcublasLt.so.11 with link time reference

1. which nvcc
   /usr/local/cuda-11.1/bin/nvcc
2. ldd /usr/local/cuda-11.1/bin/nvcc

     linux-vdso.so.1 (0x00007ffd42bb2000)
     libpthread.so.0 => /lib/x86_64-linux-gnu/libpthread.so.0 (0x00007fd18f9df000)
     libm.so.6 => /lib/x86_64-linux-gnu/libm.so.6 (0x00007fd18f641000)
     libgcc_s.so.1 => /lib/x86_64-linux-gnu/libgcc_s.so.1 (0x00007fd18f429000)
     libc.so.6 => /lib/x86_64-linux-gnu/libc.so.6 (0x00007fd18f038000)
     /lib64/ld-linux-x86-64.so.2 (0x00007fd18fbfe000)
   

3. ll /usr/local/cuda-11.1/lib64/libcublas.so
   1 root root 15 Aug 25 17:07 /usr/local/cuda-11.1/lib64/libcublas.so -> libcublas.so.11
4. echo $LD_LIBRARY_PATH
   :/usr/local/cuda-11.1/lib64
5. export CUDA_HOME=/usr/local/cuda-11.1
   export PATH=$PATH:/usr/local/cuda-11.1/bin
   export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.1/lib64
6. whereis libcudart
   libcudart: /usr/share/man/man7/libcudart.7
7. locate libcublas.so
   /home/anaconda3/pkgs/cudatoolkit-10.2.89-hfd86e86_1/lib/libcublas.so

Answer 1

可以截下图再看看