Anaconda安装Tensorflow-GPU

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

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/8996/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

免费咨询
免费咨询
扫码关注
扫码关注
联系站长

站长Johngo!

大数据和算法重度研究者!

持续产出大数据、算法、LeetCode干货,以及业界好资源!

2022012703491714

微信来撩,免费咨询:xiaozhu_tec

分享本页
返回顶部