tensorflow基础学习-anaconda方式安装tensorflow(gpu-2.5版本)

基于tensorflow1.0+太混乱了,博主转战2.5
tensorflow1.0+相应的安装过程可以见tensorflow1.15.0安装
tensorflow基础学习-anaconda方式安装tensorflow(gpu-2.5版本)_第1张图片

1、安装前确定版本信息

tensorflow基础学习-anaconda方式安装tensorflow(gpu-2.5版本)_第2张图片

2、创建虚拟环境

打开Anaconda中的 Anaconda Prompt
tensorflow基础学习-anaconda方式安装tensorflow(gpu-2.5版本)_第3张图片
输入命令创建新的虚拟环境

conda create -n TF25GPU python=3.7.0

注意安装的版本,这里的版本和上面的表格是对应的
tensorflow基础学习-anaconda方式安装tensorflow(gpu-2.5版本)_第4张图片

3、激活虚拟环境

输入命令

conda activate TF25GPU 

在服务器上输入必须加入conda
进入虚拟环境
在这里插入图片描述

4、安装CUDA

查看可供安装的版本

conda search cudatoolkit

tensorflow基础学习-anaconda方式安装tensorflow(gpu-2.5版本)_第5张图片
查看之前我要安装的tensorflow_gpu-2.15,我需要安装cuda版本是11.2的,但是没有,所以我安装11.3.1
安装cuda命令

conda install cudatoolkit=11.3.1

tensorflow基础学习-anaconda方式安装tensorflow(gpu-2.5版本)_第6张图片

这里后来在服务器gpu跑代码的时候,更换了cuda的版本是11.2
linux服务器安装cuda

5、安装cudnn

同样,先查找可安装的版本

conda search cudnn

tensorflow基础学习-anaconda方式安装tensorflow(gpu-2.5版本)_第7张图片

安装对应版本

conda install cudnn=8.2.1

tensorflow基础学习-anaconda方式安装tensorflow(gpu-2.5版本)_第8张图片

6、安装tensorflow-gpu

同样,可以查看可安装的版本

conda search tensorflow-gpu

tensorflow基础学习-anaconda方式安装tensorflow(gpu-2.5版本)_第9张图片

pip install tensorflow-gpu==2.5.0 -i https://pypi.doubanio.com/simple

7、查看我们安装的相关版本信息

conda list

tensorflow基础学习-anaconda方式安装tensorflow(gpu-2.5版本)_第10张图片

8、安装成功进行测试

python
import tensorflow as tf
tf.__version__

tensorflow基础学习-anaconda方式安装tensorflow(gpu-2.5版本)_第11张图片

你可能感兴趣的:(TensorFlow,深度学习,机器学习,机器学习,深度学习,tensorflow)