1.0 安装前的准备

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

如图我的显卡支持的最高版本为11.6
右键 打开NVIDIA面板

第二步:打开链接拉到最后 自行选取CUDA和cuDNN版本

如下图

对于我来说 我可以安装的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亲测).

来源:阿昌涩涩发逗

物联沃分享整理
物联沃-IOTWORD物联网 » Anaconda安装Tensorflow-GPU

发表评论