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

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

我下载的是"CUDA11.1+PyTorch1.9.0+..."

使用的conda命令是

cpp 复制代码
conda install pytorch=1.9.0 torchvision=0.10.0 torchaudio=0.9.0 cudatoolkit=11.1

前言

在win终端使用命令

cpp 复制代码
nvidia-smi

看最高支持的CUDA版本(我的是11.4)

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

大家可以参考这篇博文

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

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

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

cpp 复制代码
nvcc -V  

or

cpp 复制代码
nvcc --version

看CUDA安装是否成功

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

成功画面如下

2、在anaconda终端安装PyTorch等

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

cpp 复制代码
conda install pytorch=1.9.0 torchvision=0.10.0 torchaudio=0.9.0 cudatoolkit=11.1

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

执行测试程序

执行结果如下

说明安装正确

遇到的问题

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 ...

得以解决博客

暂时写这么多,有问题后面在更新...

相关推荐
yuhuofei20215 小时前
【Python入门】Python与PyCharm的安装
开发语言·python·pycharm
多年小白6 小时前
紫光国微(002049) 分析
大数据·科技·深度学习·ai
动物园猫7 小时前
金属外表多种生锈检测数据集分享(适用于YOLO系列深度学习分类检测任务)
深度学习·yolo·分类
cyyt7 小时前
深度学习周报(5.11~5.17)
人工智能·深度学习
搬砖的小码农_Sky7 小时前
如何在RX 7900 XTX显卡上运行ROCm?
深度学习·机器学习·gpu算力
RSTJ_16257 小时前
PYTHON+AI LLM DAY FOURTY-EIGHT
开发语言·人工智能·python·深度学习
Cosolar8 小时前
AI Agent 记忆机制全景对比:OpenClaw vs QwenPaw vs Hermes vs HiClaw
人工智能·深度学习·语言模型·chatgpt·面试
Gigavision10 小时前
rPPGMamba:面向 PURE-UBFC-MMPD 跨被试远程生理感知的 Mamba 时序建模方案
python·深度学习·rppg
初心未改HD10 小时前
深度学习之优化器详解
人工智能·深度学习
love530love11 小时前
ComfyUI:为什么说它是 AIGC 应用层面的集大成者?
人工智能·pytorch·windows·aigc·devops·comfyui·extensions