Anaconda安装Tensorflow-GPU
1.0 安装前的准备
第一步:查看显卡支持的CUDA版本
如图我的显卡支持的最高版本为11.6
第二步:打开链接拉到最后 自行选取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亲测).
来源:阿昌涩涩发逗