当前位置:网站首页>ImportError: /usr/local/cuda-11.2/lib64/libcublas.so.10: version `libcublas.so.10‘ not found

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

2022-08-11 08:23:00 想搞钱的小陈

首先说一下我的环境:

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版本。 

原网站

版权声明
本文为[想搞钱的小陈]所创,转载请带上原文链接,感谢
https://blog.csdn.net/qq_34845880/article/details/126262030