深度学习环境安装|PyCharm,Anaconda,PyTorch,CUDA,cuDNN等

这篇具有很好参考价值的文章主要介绍了深度学习环境安装|PyCharm,Anaconda,PyTorch,CUDA,cuDNN等。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

本文参考了许多优秀博主的博客,大部分安装步骤可在其他博客中找到,鉴于我本人第一次安装后,时隔半年,我忘记了当时安装的许多细节和版本信息,所以再一次报错时,重装花费了大量时间。因此,我觉得有必要把主要过程记录下来,以便下次需要时快速安装。以下过程不一定适用于所有人,因为它是基于我第二次安装时所遇到的问题和关键步骤的记录,谢谢大家。

我下载的是“CUDA11.1+PyTorch1.9.0+…”

使用的conda命令是

conda install pytorch=1.9.0 torchvision=0.10.0 torchaudio=0.9.0 cudatoolkit=11.1

前言

在win终端使用命令

nvidia-smi

看最高支持的CUDA版本(我的是11.4)
深度学习环境安装|PyCharm,Anaconda,PyTorch,CUDA,cuDNN等,深度学习,pycharm,pytorch

1、关于Anaconda,PyTorch和CUDA的安装

大家可以参考这篇博文。
另外我再给出几个链接,
1、pytorch和CUDA等工具包版本对照信息
2、CUDAtoolkit工具包下载
3、NVIDIA驱动程序下载
4、pytorch各版本下载(第一次安装不知道用没用;但第二次没用到,用的是conda命令安装)

首先,对于CUDA的安装,我使用2号链接下载CUDA11.1,具体流程可以参考博文(Step2和Step3,Step1视情况参考)

安装完,在win终端使用命令

nvcc -V  

or

nvcc --version

看CUDA安装是否成功

注意:如果之前失败的话,要重新打开win终端输入命令(亲测是这样的)!

注意!第一个命令的V是大写,小写会有错误出现。

成功画面如下
深度学习环境安装|PyCharm,Anaconda,PyTorch,CUDA,cuDNN等,深度学习,pycharm,pytorch

2、在anaconda终端安装PyTorch等

使用命令(版本对照信息参考链接1)

conda install pytorch=1.9.0 torchvision=0.10.0 torchaudio=0.9.0 cudatoolkit=11.1

因为前面步骤安装了CUDA11.1,所以这条命令不知道后面还要不要安装CUDA11.1,我是完整执行了这条命令,没发现错误。

执行测试程序
深度学习环境安装|PyCharm,Anaconda,PyTorch,CUDA,cuDNN等,深度学习,pycharm,pytorch
执行结果如下
深度学习环境安装|PyCharm,Anaconda,PyTorch,CUDA,cuDNN等,深度学习,pycharm,pytorch
说明安装正确

遇到的问题

1、ImportError: DLL load failed while importing _nnls: 找不到指定的模块。
得以解决博客

但我估计用不到第二步,第一步后应该就解决了

解决办法:
第一步:conda remove --force numpy, scipy
第二步:pip install -U numpy, scipy

2、UserWarning: h5py is running against HDF5 1.10.6 when it was built against 1.10.4 …

得以解决博客

暂时写这么多,有问题后面在更新…文章来源地址https://www.toymoban.com/news/detail-584777.html

到了这里,关于深度学习环境安装|PyCharm,Anaconda,PyTorch,CUDA,cuDNN等的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包