上QQ阅读APP看书,第一时间看更新
2.1 确保TensorFlow工作正常
首先,你应该确保一切都正常工作。检查汽车的油位,修理地下室保险丝,确保你的信用卡都还款了。我开玩笑的,我们讨论的是TensorFlow。
为你的第一段代码创建一个名为test.py的新文件。使用如下脚本导入TensorFlow:
import tensorflow as tf
有技术性的困难吗
如果你安装了GPU版本,而库无法搜索CUDA驱动程序,在这一步通常会出现错误。请记住,如果使用CUDA编译库,则需要使用CUDA的路径更新环境变量。检查TensorFlow上的CUDA说明。(更多信息请参见https://www.tensorflow.org/install/gpu。)
这一导入让TensorFlow准备执行你的命令。如果Python解释器没有报错,你就可以开始使用TensorFlow了。
遵守TensorFlow的惯例
导入TensorFlow库时通常使用tf
作为别名。一般来说,使用tf
来限定TensorFlow是一个好主意,因为它可以使你与其他开发人员和开源TensorFlow项目保持一致。当然,你可以使用另一个别名(或者不用别名),但是在你自己的项目中重用其他人的TensorFlow代码将是一个复杂的过程。NumPy使用np
、Matplotlib使用plt
作为别名也是如此,你将在整本书中看到它们作为约定使用。