- 博客(2)
- 收藏
- 关注
原创 Ubuntu系统 TensorFlow使用GPU训练
本教程的前提是在ubuntu系统内,已成功安装NVIDIA显卡驱动,若没有安装NVIDIA显卡驱动点这里本人亲测可用配置:ubuntu 20.04 + GTX 1080Ti + TensorFlow-gpu 2.3.1 + CUDA 10.1 + CUdnn 7.6.0 + python 3.8.10首先,不同TensorFlow版本对应的CUDA版本也是不同的。如果你安装最新或最高支持的版本,但tensorflow不支持,也没啥用。TensorFlow官网给出的测试配置在这下载地址选择适合自己的CUDA版
2023-11-16 21:09:30
432
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人