Anconda中快速部署cuda和cudnn

Anconda中快速部署cuda和cudnn,第1张

上一篇博客介绍了在本地部署GPU的方法,但是真正实践时存在很多问题,运行代码时依旧不能使用。
这里介绍一个使用conda在虚拟环境部署cuda的方法,高校便捷的让你使用gpu加速训练代码。话不多说,开始上步骤。
不管是miniconda还是anaconda都可以。
首先在命令行输入

conda create -n gpu python=3.7 cudatoolkit=11.3 cudnn

这是创建虚拟环境的第一条命令。其中需要注意的是python的环境。由于我的电脑30系显卡,所以安装的cuda的版本要大于11,好像有个代码可以查看环境里能安装的cuda版本,忘记了。我的电脑没有11.2的,所以安装11.3的。另外记住是cudatoolkit。另外cudnn的版本不需要指定,系统会自动适配。
其实问题大多出现在版本上,大家结合自己电脑尽量选择合适版本(python、cudatoolkit)。
接着我们分别安装TensorFlow和pytorch
TensorFlow的版本我们看这个网址进行匹配:添加链接描述
像cuda11.3版本直接匹配tf的最高版本,安装时不

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/langs/734709.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-04-27
下一篇 2022-04-27

发表评论

登录后才能评论

评论列表(0条)

保存