本人电脑GPU是GTX 1650,可以完美安装,应该大家都可以安装的!!!😀😀😀

1.首先打开Anaconda Prompt终端

在这里插入图片描述

2. 搭建tensorflow2.3.1 python 3.8.3环境

请输入以下代码🎈

conda create --name tensorflow2.3.1 python==3.8.3

在这里插入图片描述

  • 遇到Process,直接输入“y”,回车就好!在这里插入图片描述
3.激活tensorflow2.3.1环境

输入以下代码🎈🎈

conda activate tensorflow2.3.1

在这里插入图片描述

4.安装cudatoolkit cudnn

输入以下代🎈🎈🎈

conda install cudatoolkit=10.1 cudnn=7.6.5

在这里插入图片描述

  • 遇到Process,直接输入“y”,回车就好!在这里插入图片描述
5.安装tensorflow-gpu==2.3.1

输入以下代码,镜像源采用阿里云🎈🎈🎈🎈

pip install tensorflow-gpu==2.3.1 -i https://mirrors.aliyun.com/pypi/simple/

在这里插入图片描述

出现这样就安装好了
在这里插入图片描述

6. 检验是否可以使用🎈🎈🎈🎈🎈

以此输入以下指令,查看对应版本,ok

python
import tensorflow as tf
tf.__version__
print(tf.test.is_gpu_available())

在这里插入图片描述

参考文献:tensorflow-gpu-2.3.1安装 tensorflow安装 GPU版本tensorflow安装 tensorflow搭建

GitHub 加速计划 / te / tensorflow
184.54 K
74.12 K
下载
一个面向所有人的开源机器学习框架
最近提交(Master分支:1 个月前 )
a49e66f2 PiperOrigin-RevId: 663726708 2 个月前
91dac11a This test overrides disabled_backends, dropping the default value in the process. PiperOrigin-RevId: 663711155 2 个月前
Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐