Advertisement

在Ubuntu 16.04上安装NVIDIA CUDA 10 + cudnn 7 + PyTorch踩雷经验总结

阅读量:

近期耗费约两至三天的时间,从基础开始搭建深度学习的相关环境,过程中查阅了大量技术博客,并经历了一些问题与挑战,特此撰写本文以记录部分经验,期望能为有相同兴趣的朋友们提供一定参考与帮助。

NVIDIA显卡驱动安装:

当前可通过官方网站查询硬件所适配的版本信息,具体访问地址为:https://www.nvidia.com/Download/index.aspx?lang=en-us

在确认目标版本之后,可采用添加ppa源的方式完成软件的下载安装。

复制代码
 sudo add-apt-repository ppa:graphics-drivers/ppa  
    
 sudo apt-get update  
    
 sudo apt-get install nvidia-390 #此处要根据上面查询到的版本适当更改
    
 sudo apt-get install mesa-common-dev  
    
 sudo apt-get install freeglut3-dev
    
    
    
    

完成安装后需重新启动系统,并在终端界面中执行下列命令以开展测试工作

复制代码
    nvidia-smi  #若出现电脑GPU列表,即安装成功
    

CUDA10安装

全部评论 (0)

还没有任何评论哟~