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())