运行torch报错cuda

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

可以截下图再看看