Windows下编译带CUDA 10.2的TensorFlow 2.2(Python3.8.2,附编译结果下载)

官方教程说得基本差不多了,然而在此基础上编译还是会出问题,因此做一些补充。环境准备1. Python & Pip首先是Python需要安装一些包,除了官方说的six、numpy、wheel、keras_applications和keras_preprocessing,还需要setuptools(虽然这是个基础包,但是我习惯用虚拟环境了,忘了装……)。因此安装命令如下:pip install six numpy wheel setuptoolspip install keras_a
摘要由CSDN通过智能技术生成

官方教程说得基本差不多了,然而在此基础上编译还是会出问题,因此做一些补充。

环境准备

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

  • 7
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值