1. 未安装cuda需要先安装cuda--命令行:sudo apt install nvidia-cuda-toolkit
输入该命令行之后一般要求输入权限密码.
ps: 不清楚是否安装可输入nvcc -V来检查,如出现以下图识即表明cuda已安装成功/
2. 创建一个新环境
conda create -n xxx(命名) python=3.x(你想要的版本)
3.激活环境
conda activate xxx(刚刚的命名)
4.安装tensorflow-gpu环境
pip install tensorflow-gpu==2.x(我这里安装2.6,安装其他版本与自己的cuda版本对应即可)
5.安装cuda以及cudnn
conda install cudatoolkit=11.3
conda install cudnn=8.2
(以上版本是需要跟你安装的tensorflow版本对应的)
6.检查安装情况
--python
--import tensorflow as tf
如果出现如下错误:
即需要输入命令行:pip install protobuf==3.20
--如果没有出现错误:
![](https://i-blog.csdnimg.cn/direct/b5f8ac4505054eae9c5be74a855ffd59.png)
即安装成功