排除:
1.python编译器是64位
查询教程
2. cuda cudnn版本
均是12.2 可以向下兼容
cmd: nvcc -V即可
另一种方法
- tensorflow官网教程
pip install tensorflow_gpu==1.12.0
4.安装torch-gpu 检查所在环境
解决!!
conda install tensorflow_gpu
conda list(千万不要pip list 会显示安装了但是不是gpu版本)
import tensorflow as
print(tensorflow.test.is_gpu_available())
确实输出了TRUE;
在程序中加入
import os
os.environ[‘CUDA_VISIBLE_DEVICES’] = “0”
即可使用gpu运行
5.程序运行时出错
30系列显卡需要单独配置 nvidia的tensorflow
nvidia-tensorflow