栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

【备忘记录】Anaconda + tensorflow-gpu 安装

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

【备忘记录】Anaconda + tensorflow-gpu 安装

        为了防止每次再安装tensorflow时都要查一堆的网页和资料,我就先记录一下tensorflow-gpu的安装流程。仅仅是防止自己忘记,这里不可避免地肯定夹杂着大量别人写的东西,我本质上不过是把这些东西集中在一起罢了(简称:大融合怪)。

第一步 安装Anaconda

Anaconda | The World's Most Popular Data Science Platformhttps://www.anaconda.com/

第二步 换源

        在 C:Users用户名 文件夹下找到 .condarc 文件,以记事本形式打开,改成这样:

show_channel_urls: true
ssl_verify: true
channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  - defaults

        但是清华源经常出问题,换成别的更好。如果不进行换源这一步地话,可以在安装时每句话都加上一个 -i https://pypi.douban.com/simple/ ,这个豆瓣源用着速度和稳定性都不错。例如:

pip install tensorflow-gpu==1.15.0 -i https://pypi.douban.com/simple/
第三步 创建新环境

        从如图位置(可能别的地方也行)进入命令行:

        输入指令创建环境:

conda create -n 环境名 python=版本号

         等待各项安装,创建好环境后,激活进入环境

conda activate 环境名
第四步 安装tensorflow-gpu

        注:conda与pip的区别 conda和pip都能安装包,pip是在直接当前环境中下载安装,conda会先在一个统一放包目录下找一遍,有就直接搬过来,没有再去下载。(conda的话貌似没有 -i 参数,需要进行换源)

        可以先用命令搜寻一下再安装:

conda search tensorflow-gpu

        可以先更新pip(见问题1,为啥更新暂时不懂,不更新也行,看了下也就从21.x更到了22.x)或者直接安装(不加版本号默认最新的):

pip install tensorflow-gpu==2.6.3 -i http://pypi.douban.com/simple/

        可以用指令查看环境内安装了哪些包:

conda list

 第五步 安装cuda和cudnn

        安装哪个版本的cuda和cudnn需要按照tensorflow-gpu的版本查一下对应的!

        安装cuda,找一下可用的版本(搜索不到先换源,见步骤2或问题2):

conda search cudatoolkit

        

        tensorflow-gpu2.6.3需要cuda11.2(但是这里没有,11.3也可以):

conda install cudatoolkit=11.3

         查看可安装cudnn版本:

conda search cudnn

        安装cudnn8.2.1:

conda install cudnn=8.2.1

         查看安装结果:

第六步 到jupyter里进行测试
import tensorflow as tf
tf.__version__
tf.test.is_gpu_available()
tf.config.list_physical_devices('GPU')

还有一个测试代码(能正确识别cpu与gpu即可):

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

  各种乱七八糟的问题:

1. 更新pip

        可以用如下命令更新(--user是管理员权限,最好指定源):

pip install --user --upgrade pip -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com

        如果没弄好把什么文件弄丢了的话就重新安装pip:

python -m ensurepip --default-pip

2. 命令行换源

        主要就是几个指令,添加指令(具体有哪些源百度一下就能找到):

conda config --add channels https://xxxxxxxxxx

        清理搜索缓存:

conda clean -i

         移除指令:

conda config --remove channels https://xxxxxxxxxx

        查看指令:

conda config --show channels

3. 后续有新问题再加

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/886532.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号