Anaconda安装Tensorflow的GPU版

发布于:2021-10-17 19:20:01

最*因为各种原因之前的tensorflow发现不能用用了,所以在网上找了这篇优质的安装指南,分享一下。


1.安装Anaconda

官网下载
地址:https://www.anaconda.com/products/individual


2. 卸载电脑上已有的python

先看下自己电脑*沧暗膒ython版本(为了以防后面的一些奇怪的版本不兼容问题)


直接在官网找到相应版本的安装包(记住是安装包,不是直接解压就用的那种),然后运行uninstall.


3. 开搞conda

一路next,到这一步后直接选择添加到path中,这样就不用自己去添加了


打开cmd看一下python版本:


python -V (这里如果单输入python的话是进入python环境,可以看到这个python是在conda环境下的)


4. 安装Tensorflow-gpu

安装GPU版本必须把对应的版本全部校对好!!!(python版本,cuda版本,cudnn版本以及tensorflow版本)


打开conda的prompt


第一步:查看版本。

conda -V


这里要更改一下国内的镜像源


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/
conda config --set show_channel_urls yes


然后在新建一个虚拟环境:


conda create -n TC2 python=3.7


进入该环境:


conda activate TC2


有这个显示就说明我们进入了这个TC2的环境,之后我们就在这个环境下安装tensorflow-gpu


conda install tensorflow-gpu //这里也可以指定版本,比如conda install tensorflow-gpu==2.1.0


输入y确认。我们不需要再额外安装cuda和cudnn,因为他已经包含在安装的列表里了。


我这里安装的是2.1.0的版本,静静等待安装完成,如果安装出现问题,比如进度卡住了之类的多半是网络不稳定,可以选择切换到国内镜像。


等待完成。然后我们可以看一下有没有安装完成:


conda list //查看当前环境下安装的包


第二步:验证

我们新建一个python3的脚本,输入:


import tensorflow as tf
tf.version


import tensorflow as tf
print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices(‘GPU’)))


本文按照以下作者内容进行删改
本文作者:陆黎的Blog
本文链接:https://www.cnblogs.com/jshmztl/p/13306837.html

相关推荐

最新更新

猜你喜欢