基本参照我的这篇文章:《Windows下编译带CUDA 10.2的TensorFlow 2.2(Python3.8.2,附编译结果下载)》,有些地方有所改动,重新组织一下步骤。
环境准备
1. Python & Pip
首先Python需要安装一些包:six、numpy、wheel、setuptools、keras_applications和keras_preprocessing,使用管理员权限打开命令提示符:
pip install six numpy wheel setuptools
pip install keras_applications --no-deps
pip install keras_preprocessing --no-deps
注意,Python路径中不能出现空格,即Windows下默认安装路径C:\Program Files\Python38会在编译时报错,因此如果装到了这个路径,需要在一个没有空格的目录下创建一个链接(不是快捷方式),用mklink命令。
2. CUDA
这里选的CUDA 11,CUDA官网下载安装,没什么好说的。
3.Bazel
然后是Bazel,bazel很简单,就一个exe,需要设置环境变量给到Path下,我偷懒直接放到CUDA的bin目录下。我选的版本是3.4.1。
4. MSYS2
再安装MSYS2,同样需要给msys64\usr\bin目录设置环境变量。
装好后再安装一些包,用的是pacman,由于默认源极慢极慢,所以建议国内换清华源。
进到msys64\etc\pacman.d目录下,修改三个mirrolist,分别在各自所有Server行前加一行:
mirrorlist.msys:
Server = https://mirrors.tuna.tsinghua.edu.cn/msys2/msys/$arch
mirrorlist.mingw32:
Server = https://mirrors.tuna.tsinghua.edu.cn/msys2/mingw/i686
mirrorlist.mingw64:
Server = https://mirrors.tuna.tsinghua.edu.cn/msys2/mingw/x86_64
官方教程少提了一个zip包,因此安装命令如下:
pacman -S git patch unzip zip
5. Visual Studio 2019
然后是VS,下载VS安装器,为避免麻烦,装到C盘默认路径(这次我没有尝试非C盘路径,不知道找不到编译器的bug还在不在)。如果非VS用户,只需安装除必选组件外的MSVC v142 - VS 2019 C