首页 > 系统相关 >windows10用conda搭建tensorflow的gpu环境

windows10用conda搭建tensorflow的gpu环境

时间:2024-07-02 15:55:56浏览次数:25  
标签:windows10 conda pkgs tensorflow gpu main 安装 2.10

在 tensorflow 官方网址上也列举了很多方法,但都很麻烦,包括 docker也没有办法在win10下应用 gpu来计算。
记录我的检查过程。

在官网搜集有用的资料。

  • “在 Windows 环境中从源代码构建”中提到了经过测试后,可用的配套版本,找到一个最新的是:
    | 版本 | Python 版本 | 编译器 | 构建工具 | cuDNN | CUDA |
    | -------------------- | --------- | --------- | ----------- | ----- | ---- |
    | tensorflow_gpu-2.6.0 | 3.6-3.9 | MSVC 2019 | Bazel 3.7.2 | 8.1 | 11.2 |
    从这条信息中,得出:tf2.6,python3.9,cuDNN8.1,CUDA11.2,这个组合是一个可靠的配套。
  • “使用 pip 安装 TensorFlow”一节中提到:软件包位置一栏中,列出很多编译过的whl文件,找到一条我感兴趣的信息:
    | Python 3.9(支持 GPU) | https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-2.6.0-cp39-cp39-win_amd64.whl |
    这个好像能和上面对应上。其中提到文件:tensorflow_gpu-2.6.0-cp39-cp39-win_amd64.whl,这个名字的含义:
    这个文件名 tensorflow_gpu-2.6.0-cp39-cp39-win_amd64.whl 是一个用于安装 TensorFlow 的 Python wheel 包。表示这是一个用于 64 位 Windows 系统的、适用于 Python 3.9 的、支持 GPU 的 TensorFlow 2.6.0 版本的安装包。
  • 还是在 pip 安装这一节也提到了使用 conda 安装,网址是:https://docs.anaconda.com/working-with-conda/applications/tensorflow/,里面提到,可以用包名 tensorflow-gpu 来搜索安装,同时也提到目前(2024.7)最新支持的tensorflow版本是2.10版。
    根据以上可以推测出,在win10下,可以安装python3.9版本,tensorflow2.6的 gpu 稳定版本,及beta版本 2.10 版本。

安装前的准备工作

  1. 要安装 gpu 环境,一个 nvidia 的显卡及驱动是不能少的,也是物理条件。在官网根据提示下载即可。
  2. 安装 miniconda。
  3. 用 miniconda 创建一个 python3.9的虚拟环境:conda create --name tf python=3.9
  4. 激活 tf 环境后,搜索 tensorflow-gpu
(tf) PS C:\Users\Logic> conda search tensorflow-gpu
Loading channels: done
# Name                       Version           Build  Channel
tensorflow-gpu                 1.8.0      h21ff451_0  pkgs/main
tensorflow-gpu                 1.9.0      hf154084_0  pkgs/main
tensorflow-gpu                1.10.0      hf154084_0  pkgs/main
tensorflow-gpu                1.11.0      h0d30ee6_0  pkgs/main
tensorflow-gpu                1.12.0      h0d30ee6_0  pkgs/main
tensorflow-gpu                1.13.1      h0d30ee6_0  pkgs/main
tensorflow-gpu                1.14.0      h0d30ee6_0  pkgs/main
tensorflow-gpu                1.15.0      h0d30ee6_0  pkgs/main
tensorflow-gpu                 2.0.0      h0d30ee6_0  pkgs/main
tensorflow-gpu                 2.1.0      h0d30ee6_0  pkgs/main
tensorflow-gpu                 2.3.0      he13fc11_0  pkgs/main
tensorflow-gpu                 2.5.0      h17022bd_0  pkgs/main
tensorflow-gpu                 2.6.0      h17022bd_0  pkgs/main

和前面在官网搜到的信息一致,看来这就是官网能够支持的版本了,要想安装最新的 2.10 需用 conda-forge 的频道。
稳定版安装命令:
conda install tensorflow-gpu=2.6.0 cuda
5. 再用 tensorflow 搜索一些看看

... 部分
tensorflow                     2.9.1 mkl_py38h7f03810_0  pkgs/main
tensorflow                     2.9.1 mkl_py38hff71f30_1  pkgs/main
tensorflow                     2.9.1 mkl_py39hb9887a6_0  pkgs/main
tensorflow                     2.9.1 mkl_py39hc9ebea8_1  pkgs/main
tensorflow                    2.10.0 eigen_py310h1d93ef9_0  pkgs/main
tensorflow                    2.10.0 eigen_py37h0b514e4_0  pkgs/main
tensorflow                    2.10.0 eigen_py38h465b00b_0  pkgs/main
tensorflow                    2.10.0 eigen_py39hfced716_0  pkgs/main
tensorflow                    2.10.0 gpu_py310h4d27e69_0  pkgs/main
tensorflow                    2.10.0 gpu_py37h5d22f32_0  pkgs/main
tensorflow                    2.10.0 gpu_py38h51559ff_0  pkgs/main
tensorflow                    2.10.0 gpu_py39h9bca9fa_0  pkgs/main
tensorflow                    2.10.0 mkl_py310hd99672f_0  pkgs/main
tensorflow                    2.10.0 mkl_py38ha5c4042_0  pkgs/main
tensorflow                    2.10.0 mkl_py39ha510bab_0  pkgs/main

找到其中的支持gpu最新的版本:tensorflow 2.10.0 gpu_py39h9bca9fa_0 pkgs/main
安装:
conda install tensorflow=2.10.0=gpu_py39h9bca9fa_0 cuda
每个安装命令都加上 cuda 是让 conda 自动安装对应的 cuda 环境,从最终安装的 cudnn=8.2 也和前面查询的版本差不多,说明的确是最佳组合了。

总结

在 tensorflow 官网没有提供像 pytorch 那样明确的安装命令,只能靠搜索分析一些数据,然后再合理推测,找到合理的安装方法。
虽然最后的安装命令就一行,但我记录下整个查找过程,记录学习。

标签:windows10,conda,pkgs,tensorflow,gpu,main,安装,2.10
From: https://www.cnblogs.com/litifeng/p/18279994

相关文章

  • anaconda使用方法
    anaconda介绍Anaconda可以管理环境和软件包,能够便捷获取包且对包能够进行管理,包括了python和很多常见的软件库和一个包管理器conda。常见的科学计算类的库都包含在里面了,使得安装比常规python安装要容易,同时对环境可以统一管理的发行版本。常用指令创建环境:condacreate--n......
  • mac M1 安装miniconda3
     进入国内大学镜像库下载安装包https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda 下载包打开终端,开始安装1、执行命令:bashMiniconda3-latest-MacOSX-arm64.sh2、输入回车后,阅读文件,输入:q退出文件 3、输入yes开始安装4、输入回车选择安装路径-默认路径/U......
  • 动手学深度学习5.6 GPU-笔记&练习(PyTorch)
    以下内容为结合李沐老师的课程和教材补充的学习笔记,以及对课后练习的一些思考,自留回顾,也供同学之人交流参考。本节课程地址:17使用和购买GPU【动手学深度学习v2】_哔哩哔哩_bilibili本节教材地址:5.6.GPU—动手学深度学习2.0.0documentation(d2l.ai)本节开源代码:...>d......
  • GPU算力租用平台推荐
    在选择GPU算力租用平台时,需要考虑多个因素,包括平台的稳定性、服务质量、价格、GPU规格和性能,以及是否满足特定的计算需求等。以下是一些值得推荐的GPU算力租用平台,以及它们的特点和优势:AWS(AmazonWebServices)提供多种GPU实例,包括TeslaV100、T4、K80等,适用于深度学习、......
  • tensorflow-gpu配置
    1.安装Anaconda下载地址:Anaconda|TheOperatingSystemforAI2.查询tensorflow-gpu不同版本所对应的python、cuDNN和CUDA的版本官网:在Windows环境中从源代码构建 | TensorFlow(google.cn) 3.使用conda安装相应的库#tensorflow_gpu-2.3.0condainitcondacrea......
  • 10分钟安装好torch的GPU版本(Windows)
    pytorch-gpu1.确定cuda版本2.确定Python版本3开始下载-cu118-cp383.1下载cuda3.2下载torchvision4.下载好了5.开始安装6.开始验证1.确定cuda版本nvcc-V版本为11.8,一会下载的版本为cu1182.确定Python版本确定python版本为为3.8,一会下载为cp38......
  • Conda环境导出指南:YAML文件的奥秘
    标题:Conda环境导出指南:YAML文件的奥秘摘要Conda不仅是一款强大的包管理器,也是Linux、macOS和Windows平台上卓越的环境管理工具。将Conda环境导出为YAML文件是一种保存环境配置的便捷方式,便于分享、迁移或重建环境。本文将详细介绍如何将Conda环境导出为YAML文件,并探讨其应......
  • FireFox 编译指南2024 Windows10篇-环境准备(一)
    1.引言在开源浏览器项目中,Firefox因其高性能和灵活性而备受开发者青睐。为了在本地环境中编译和定制Firefox,开发者需要做好充分的环境准备工作。这不仅是编译成功的基础,也是后续调试、优化和二次开发的关键步骤。编译Firefox是一个复杂而耗时的过程,涉及大量的代码文件和依赖......
  • 将Windows11右键菜单修改为Windows10风格
    Windows11更新后,右键菜单很多功能隐藏起来了,使用时需要点击“显示更多选型”才能获取完整功能。为了能获得Windows10右键菜单丝滑的体验,我总结了以下方法。方法一:控制台操作法按住win+R打开运行窗口输入cmd,打开控制台在控制泰中输入下方代码,回车,显示操作成功开启旧版右......
  • [本科项目实训] Anaconda 环境迁移
    问题描述由于本地显卡GeForceRTX4070Ti推理速度较慢,我们拟采用服务器GeForceRTX3090进行最后的项目展示,因而需要重新配置环境。由于Cuda版本和Conda版本基本一致,拟采用condapack指令进行快速迁移。解决方案第一步,在base环境下载condapack:pipinstallconda-pack第......