在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)
还没有任何评论哟~
