NVIDIA系列显卡做深度学习,需要在宿主机安装显卡驱动(driver),需要在docker中安装英伟达的科学计算库(cuda),在python中安装深度学习库,driver依赖显卡,cuda依赖driver,pytorch依赖cuda,于是就会有一些版本依赖问题.
显卡、驱动、cuda关系
最新推荐文章于 2024-04-06 19:32:29 发布
本文讲述了在使用NVIDIA系列显卡进行深度学习时,如何在宿主机上安装显卡驱动并在Docker中配置CUDA和PyTorch,特别强调了不同软件之间的版本依赖问题及其解决方法。
摘要由CSDN通过智能技术生成