环境:windows10,无需提前配置cuda。请注意我的GPU为nvidia GPU,其他厂商GPU无法用此流程。
目录
1. 安装mambaforge
2. 配置mambaforge
a) 配置PATH
b) 配置清华源
c) 配置windows支持长地址
3. 安装tensorflow和keras
4. 检验安装是否成功
1. 安装mambaforge
Conda是一个方便的python虚拟环境+包管理工具,但是使用起来比较缓慢,尤其是solving environment会花费大量时间。Mamba则是一个对conda功能的覆盖升级,大幅提高了下载和求解环境的速度。
较老一点的教程会建议通过conda安装mamba,但在windows平台上,这一操作非常耗时,且不断地失败。作为替代,可以直接下载mambaforge的exe安装文件。
下载地址为:GitHub - conda-forge/miniforge: A conda-forge distribution.
我下载安装包文件名是Mambaforge-Windows-x86_64.exe。
下载后直接双击安装即可。
2. 配置mambaforge
a) 配置PATH
下载后打开cmd,尝试运行 mamba info。
如果提示找不到mamba命令,请将mamba可执行文件加入windows path。可执行文件在安装目录下的 \condabin 目录下,在我的机器上是
C:\Users\myusername\AppData\Local\mambaforge\condabin
配置以后,从cmd运行mamba info,会有相关信息打印。
b) 配置清华源
由于默认的conda下载源在国外,从国内下载经常遇到超时失败。我们将下载源指定为清华镜像以加速下载。
执行以下命令:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
# 以上两条是Anaconda官方库的镜像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
# 以上是Anaconda第三方库 Conda Forge的镜像
conda config --set show_channel_urls yes
c) 配置windows支持长地址
tensorflow的目录非常深,部分文件地址长度超出了windows默认支持的256字符,会导致安装时报错。
用管理员权限打开PowerShell,执行以下命令:
New-ItemProperty -Path "HKLM:\SYSTEM\CurrentControlSet\Control\FileSystem" ` -Name "LongPathsEnabled" -Value 1 -PropertyType DWORD -Force
然后重启,windows即可支持长路径。
3. 安装tensorflow和keras
打开cmd,创建一个新环境:
Mamba create –n tf_gpu
激活环境:
Mamba activate tf_gpu
安装tensorflow for gpu:
Mamba install tensorflow-gpu
即会自动为您安装合适的cuda版本以及TensorFlow版本。
等待完成以后,安装keras:
Mamba install keras
然后就配置完成了。
注:另有说法是,直接install keras-gpu即可安装keras+tensorflow,但是我没有实际尝试过,不确定可行。
4. 检验安装是否成功
在tf_gpu环境中,运行python,执行以下脚本:
import tensorflow as tf
gpu_device_name = tf.test.gpu_device_name()
print(gpu_device_name)
如果没有报错,正常打印出gpu名字,则说明环境配置成功。
有任何疑问或建议,欢迎交流!