- 博客(2)
- 收藏
- 关注
原创 【ShuQiHere】在Ubuntu上安装CUDA和cuDNN
在Ubuntu系统上,正确安装和配置CUDA、cuDNN,以及TensorFlow和PyTorch,以充分利用NVIDIA GPU进行深度学习任务至关重要。本文详细讲解了如何检查NVIDIA驱动对CUDA的支持,安装兼容的CUDA版本,以及安装与CUDA匹配的cuDNN版本,并提供了避免常见问题的建议。通过这些步骤,你可以确保TensorFlow和PyTorch能够正确识别和使用GPU进行训练。
2024-08-20 12:29:44
420
原创 【ShuQiHere】在Ubuntu上安装NVIDIA驱动
本文详细介绍了如何在Ubuntu上禁用Nouveau驱动程序并安装NVIDIA官方驱动。通过逐步检查和禁用Nouveau驱动,避免了与NVIDIA驱动的冲突,确保了安装的顺利进行。最后,还提供了查询当前图形模式的方法。如果在CUDA或cuDNN的安装过程中遇到问题,可以参考下一篇文章中的详细指导。
2024-08-20 11:52:24
433
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人