Anaconda安装Tensorflow-GPU

人工智能32

1.0 安装前的准备

第一步:查看显卡支持的CUDA版本

如图我的显卡支持的最高版本为11.6
Anaconda安装Tensorflow-GPU

Anaconda安装Tensorflow-GPU

如下图
Anaconda安装Tensorflow-GPU
对于我来说 我可以安装的CUDA是11.6以下的所有版本
我选择了 10.1CUDA,则对应的 7.6cuDNN以及 2.3tensorflow

; 2.0 开始安装

首先创建一个虚拟环境

我选择2.3tensorflow+3.8python

conda create -n tf2.3 python=3.8

然后激活环境进行下一步安装(切记!)

conda install cudatoolkit=10.1
conda install cudnn=7.6
pip install tensorflow==2.3

验证是否安装成功

python
import tensorflow as tf
tf.__version__

验证GPU是否可用

import tensorflow as tf
tf.config.list_physical_devices('GPU')

[PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]

参考文章

conda安装tensorflow(python).

anaconda安装TensorFlow GPU版本(2021.11 3060亲测).

Original: https://blog.csdn.net/m0_56023586/article/details/123489752
Author: 阿昌涩涩发逗
Title: Anaconda安装Tensorflow-GPU

相关文章
机器学习中 F1-score 人工智能

机器学习中 F1-score

整合了两个链接的知识点,把里面的小错误改掉了: 机器学习中的F1-score 【深度学习笔记】F1-Score 一、定义 F1分数(F1-score)是分类问题的一个衡量指标。一些多分类问题的机器学习...
gensim实现TF-IDF和LDA模型、sklearn实现聚类 人工智能

gensim实现TF-IDF和LDA模型、sklearn实现聚类

假如我有100篇文章,每篇文章形如: 本发明公开的一种冰箱用的三风门,其包括:一风门本体,所述风门本体上设置有第一风口、第二风口和第三风口;安装在所述风门本体上的电动机构;分别设置在所述第一风口、第二...
QT实现语音识别功能 人工智能

QT实现语音识别功能

QT实现语音识别功能的相关步骤如下(附程序代码): 1.阿里云注册账号,登录智能语音交互平台:创建新项目,此时或获得AppKey,进入智能语音交互总览一页,右上方会看见一个Access Token,然...
人工智能笔记 人工智能

人工智能笔记

第一章:绪论 1956年正式提出人工智能(artificial intelligence, AI)这个术语并把它作为一门新兴科学的名称。 20世纪三大科学技术成就:空间技术、原子能技术、人工智能 智能...