2021-07-12 在 GeForce RTX 3090上配置深度学习环境 cuda 11.1 + tensorflow2.5.0 + python3.8.3
GeForce RTX 3090
配置环境的过程遇到了很多问题,最后成功配置的版本如下
tensorflow-gpu 2.5.0
cudnn 8.1.0.77
python 3.8.3
cuda 11.1
参考的版本对应关系如图
成功安装的细节
安装tensorflow-gpu 2.5.0
conda activate 虚拟环境名字
pip install tensorflow-gpu==2.5.0 # conda install tensorflow-gpu=2.5.0 会找不到
原创
2021-07-13 00:52:40 ·
3880 阅读 ·
6 评论