Anaconda Prompt中装tensorflow_gpu

2024-01-22 03:30

本文主要是介绍Anaconda Prompt中装tensorflow_gpu,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

以tensorflow_gpu-1.13.1为例:

下载anaconda:Index of /anaconda/archive/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

 

一、anaconda建立虚拟环境

1.先建立虚拟环境一个虚拟环境(在anaconda powershell prompt base中)

conda create -n tf_1x python==3.6  

(python可写上为了指定版本)

2.激活虚拟环境

(base) PS C:\Users\hp> conda activate tf_1x

二、.配置相应的cuDNN版本和CUDA版本

   注意配置:

   python=2.7,3.3~3.6 ;本人下载3.6

   cuDNN版本:≥7.4 ,<8 ;本人下载7.6.5.32版本,有7.4版本最好(越接近越好)

   CUDA版本:≥10.0,<11 ;本人下载10.0.130版本,有10.0版本最好(越接近越好)

 (tf_1x) PS C:\Users\hp> conda install cudatoolkit=10.0.130 cudnn=7.6.5.32

如果不知道cuDNN版本和CUDA版本匹配,先进入虚拟环境之后可以按照一下进行操作

(1)搜索库名:cudatoolkit,命令如下:

anaconda search -t conda cudatoolkit

如图

(2)查看包的详细信息,选择要安装版本,过程中选择:y,命令如下:

anaconda show conda-forge/cudatoolkit

(3)本人安装cudatoolkit 10.0.130版本,命令如下:

conda install --channel https://conda.anaconda.org/conda-forge cudatoolkit=10.0.130

   等待安装完成。

同理,搜索cuDNN选择相应版本,命令安装,命令如下:

anaconda show conda-forge/cudnn
conda install --channel https://conda.anaconda.org/conda-forge cudnn=7.6.5.32

三、安装完成后,开始安装tensorflow 1.13.1

命令如下:

pip install tensorflow-gpu==1.13.1 -i https://mirrors.aliyun.com/pypi/simple/

此处用aliyun镜像源下载,可以加快速度。

四、检查是否成功安装在GPU上

在python窗口或者pycharm中(pycharm中需要先添加虚拟环境,添加步骤请参考下一篇文章在pycharm中添加虚拟环境_m0_63172128的博客-CSDN博客)

命令如下:

import tensorflow as tf
print(tf.__version__)

五、附加可能用到的相关代码:

关闭虚拟环境:

deactivate yourname

移除虚拟环境:

conda remove -n yourname --all

删除环境中的某个包:

conda remove --name yourname(虚拟环境名) name(包名)

可以返回本地源,退出镜像源

conda config --remove-key channels

六、总结

1、在anaconda powershell prompt base中用相关命令先建立虚拟环境(相关代码参考本文详情)

2、继续激活建立的虚拟环境(相关代码参考本文详情)

3、考虑好自己要安装的tensorflow版本,去官网参考对应的cuDNN版本和CUDA版本。

4、安装cuDNN版本和CUDA(相关代码参考本文详情)

5、下载tensoreflow(相关代码参考本文详情)

6、测试安装是否成功(相关代码参考本文详情)

附:本人亲试安装成功可用,有不足地方欢迎批评指正!

这篇关于Anaconda Prompt中装tensorflow_gpu的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/631703

相关文章

AI Toolkit + H100 GPU,一小时内微调最新热门文生图模型 FLUX

上个月,FLUX 席卷了互联网,这并非没有原因。他们声称优于 DALLE 3、Ideogram 和 Stable Diffusion 3 等模型,而这一点已被证明是有依据的。随着越来越多的流行图像生成工具(如 Stable Diffusion Web UI Forge 和 ComyUI)开始支持这些模型,FLUX 在 Stable Diffusion 领域的扩展将会持续下去。 自 FLU

如何用GPU算力卡P100玩黑神话悟空?

精力有限,只记录关键信息,希望未来能够有助于其他人。 文章目录 综述背景评估游戏性能需求显卡需求CPU和内存系统需求主机需求显式需求 实操硬件安装安装操作系统Win11安装驱动修改注册表选择程序使用什么GPU 安装黑神话悟空其他 综述 用P100 + PCIe Gen3.0 + Dell720服务器(32C64G),运行黑神话悟空画质中等流畅运行。 背景 假设有一张P100-

Prompt - 将图片的表格转换成Markdown

Prompt - 将图片的表格转换成Markdown 0. 引言1. 提示词2. 原始版本 0. 引言 最近尝试将图片中的表格转换成Markdown格式,需要不断条件和优化提示词。记录一下调整好的提示词,以后在继续优化迭代。 1. 提示词 英文版本: You are an AI assistant tasked with extracting the content of

Anaconda 中遇到CondaHTTPError: HTTP 404 NOT FOUND for url的问题及解决办法

最近在跑一个开源项目遇到了以下问题,查了很多资料都大(抄)同(来)小(抄)异(去)的,解决不了根本问题,费了很大的劲终于得以解决,记录如下: 1、问题及过程: (myenv) D:\Workspace\python\XXXXX>conda install python=3.6.13 Solving environment: done.....Proceed ([y]/n)? yDownloa

GPU 计算 CMPS224 2021 学习笔记 02

并行类型 (1)任务并行 (2)数据并行 CPU & GPU CPU和GPU拥有相互独立的内存空间,需要在两者之间相互传输数据。 (1)分配GPU内存 (2)将CPU上的数据复制到GPU上 (3)在GPU上对数据进行计算操作 (4)将计算结果从GPU复制到CPU上 (5)释放GPU内存 CUDA内存管理API (1)分配内存 cudaErro

Python知识点:如何使用Anaconda进行科学计算环境管理

使用 Anaconda 进行科学计算环境管理是一个非常强大且灵活的方式,特别适合处理 Python 和 R 语言的包管理和虚拟环境管理。Anaconda 集成了许多用于科学计算和数据分析的库,并提供了环境隔离的功能,确保不同项目之间不会发生包冲突。以下是使用 Anaconda 进行科学计算环境管理的详细步骤: 1. 安装 Anaconda 首先,你需要在本地机器上安装 Anaconda。你可以

PyInstaller问题解决 onnxruntime-gpu 使用GPU和CUDA加速模型推理

前言 在模型推理时,需要使用GPU加速,相关的CUDA和CUDNN安装好后,通过onnxruntime-gpu实现。 直接运行python程序是正常使用GPU的,如果使用PyInstaller将.py文件打包为.exe,发现只能使用CPU推理了。 本文分析这个问题和提供解决方案,供大家参考。 问题分析——找不到ONNX Runtime GPU 动态库 首先直接运行python程序

win10不用anaconda安装tensorflow-cpu并导入pycharm

记录一下防止忘了 一、前提:已经安装了python3.6.4,想用tensorflow的包 二、在pycharm中File-Settings-Project Interpreter点“+”号导入很慢,所以直接在cmd中使用 pip install -i https://mirrors.aliyun.com/pypi/simple tensorflow-cpu下载好,默认下载的tensorflow

麒麟系统安装GPU驱动

1.nvidia 1.1显卡驱动 本机显卡型号:nvidia rtx 3090 1.1.1下载驱动 打开 https://www.nvidia.cn/geforce/drivers/ 也可以直接使用下面这个地址下载 https://www.nvidia.com/download/driverResults.aspx/205464/en-us/ 1.1.3安装驱动 右击,

稀疏自编码器tensorflow

自编码器是一种无监督机器学习算法,通过计算自编码的输出与原输入的误差,不断调节自编码器的参数,最终训练出模型。自编码器可以用于压缩输入信息,提取有用的输入特征。如,[1,0,0,0],[0,1,0,0],[0,0,1,0],[0,0,0,1]四比特信息可以压缩成两位,[0,0],[1,0],[1,1],[0,1]。此时,自编码器的中间层的神经元个数为2。但是,有时中间隐藏层的神经元