官方教程
说得基本差不多了,然而在此基础上编译还是会出问题,因此做一些补充。
环境准备
1. Python & Pip
首先是Python需要安装一些包,除了官方说的six、numpy、wheel、keras_applications和keras_preprocessing,还需要setuptools(虽然这是个基础包,但是我习惯用虚拟环境了,忘了装……)。因此安装命令如下:
pip install six numpy wheel setuptools
pip install keras_applications==1.0.6 --no-deps
pip install keras_preprocessing==1.0.5 --no-deps
注意,Python路径中不能出现空格,即Windows下默认安装路径
C:\Program Files\Python38
会在编译时报错,因此如果装到了这个路径,需要在一个没有空格的目录下创建一个链接(不是快捷方式),用mklink命令。
2. CUDA
这里选的CUDA 10.2,CUDA官网下载安装,没什么好说的。
不过需要注意,TensorFlow 2.2配置时能正确找到CUDA 10.2的路径,编译时却只会去找CUDA_PATH_V10_1这个环境变量,需要手动在系统环境变量里指定这个到CUDA 10.2的路径(跟C
版权声明:本文为u012440550原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。