《新手向》CUDA和CUDNN新手安装需要注意的坑

CUDA和CUDNN是加速深度学习用的,一般涉及到GPU的项目都会需要提前装好它,它是英伟达公司出品的,因此仅仅适用于英伟达的显卡GPU。

例如说很多人在python中第一次装pytorch时,会出现报错可能就是因为你装了GPU版的Pytorch但是没有装CUDA。此时你可以去浏览器搜索到pytorch的cpu版本进行安装。

坑人点:

1,cuda官方版本一定不要用最新的。大部分项目不会对cuda最新版做好适配,日后项目运行可能会出现无法解决的问题,而且你还一脸懵找到不到原因。

2,安装cuda和cudnn的时候一定要提前看一下项目说明或依赖文件,看一下需要你装哪个版本的,装一次不容易别搞错了。

3,cudnn是cuda的配套的组件,也要装,安装方法就是下载一个文件夹,然后手动把文件移到cuda的目录下,因为不是用程序装的,因此要手动配置好环境变量。多个文件配置多个环境变量。不要遗漏。

4,如果你要装的Pytorch,请注意好CUDA的版本,举个例子torch9.9版本对应cuda8.5版本,版本要对应好,其他需要注意版本的也是要注意。

4,除了这些坑,其余的安装教程你都可以在CSDN中找到,教程很多但很少有人分享实际安装的问题。如果对你有用,不妨为我点个赞,也欢迎收藏,随时观看。

猜你喜欢

转载自blog.csdn.net/qq_45366830/article/details/129997710