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亲测）.
全部评论 (0)
还没有任何评论哟~