ImportError: /usr/local/cuda-11.2/lib64/libcublas.so.10: version `libcublas.so.10‘ not found

首先说一下我的环境:

cuda==11.2

pytorch==1.12.1

tensorflow==2.6.0

1.首先查看了cuda的运行版本,

nvcc --version   查看cuda版本

发现是11.2的,正常;

2.查看tensorflow

import tensorflow as tf 

tf.test.is_gpu_aviauable() 

发现返回的true,说明环境没问题。

3.最后定位问题再pytorch的版本上,换了一个版本,升到1.8.1的版本就可以了,具体原因我猜测是由于1.12.1不支持11.2的cuda版本。 

你可能感兴趣的:(Bug,深度学习,tensorflow,python)