首先说一下我的环境:
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版本。