当我们在使用pip安装tensorflow时,有时会遇到模块缺失的错误,例如:
ImportError: No module named tensorflow.python.
这是因为我们在安装tensorflow的过程中,系统会默认选择tensorflow的CPU版本,而忽略了我们系统中是否存在GPU加速库CUDA和cudnn。因此我们需要手动指定安装tensorflow的GPU版本,并且安装相关的依赖库。
以下是解决这个问题的攻略:
1. 检查CUDA和cudnn是否安装成功
先检查一下我们系统中是否已经安装了CUDA和cudnn。我们可以通过以下命令来检查:
$ nvcc -V
如果已经安装了CUDA,这个命令会输出相应的CUDA版本信息。
接下来检查cudnn,我们可以到cudnn官网 https://developer.nvidia.com/cudnn 下载和安装。安装完成后,我们可以在/usr/local/cuda/include
和 /usr/local/cuda/lib64
目录下找到相关文件,确认安装是否成功。
2. 重新安装tensorflow-gpu
我们需要手动安装GPU版本的tensorflow,命令如下:
pip install tensorflow-gpu
安装完成后,我们可以在Python环境下输入以下代码来验证是否安装成功:
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
如果没有报错,我们也可以查看一下tensorflow调用的运算设备:
Device mapping:
/job:localhost/replica:0/task:0/device:GPU:0 -> device: 0, name: GeForce GTX 1080, pci bus id: 0000:82:00.0, compute capability: 6.1
以上是解决pip安装tensorflow中出现的No module named 'tensorflow.python'
问题的攻略,希望能对大家有所帮助。
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:解决pip安装tensorflow中出现的no module named tensorflow.python 问题方法 - Python技术站