Windows下编译带CUDA 11.1(Update 1)的TensorFlow 2.4(RC0)(Python3.9.0,cuDNN 8.0.4,兼容性3.5 - 8.6,附编译结果下载)

 基本参照我的这篇文章:《Windows下编译带CUDA 11的TensorFlow 2.3(Python3.8.5,cuDNN 8.0.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\Python39会在编译时报错,因此如果装到了这个路径,需要在一个没有空格的目录下创建一个链接(不是快捷方式),用mklink命令。

2. CUDA

这里选的CUDA 11.1 Update 1,CUDA官网下载安装,没什么好说的。

3. Bazel

然后是Bazel,bazel很简单,就一个exe,需要设置环境变量给到Path下,我偷懒直接放到CUDA的bin目录下。我选的版本是3.7.0。

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
  • 0
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 4
    评论
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值