Anaconda+Cuda+Cudnn+Pytorch(GPU版)+Pycharm+Win11深度学习环境配置

简介: Anaconda+Cuda+Cudnn+Pytorch(GPU版)+Pycharm+Win11深度学习环境配置

一、准备工作

个人电脑配置:RTX4060 win11

个人配置版本:cuda(11.7)+ pytorch(2.0.1) + python(3.9)

所需工具:

1、python集成开发环境:Anaconda

2、CUDA、cuDNN:英伟达提供的针对英伟达显卡的运算平台。用来提升神经网络的运行效率,如果电脑显卡不满足要求也是可以不用安装,使用cpu来进行运算。

3、开发工具:PyCharm

4、深度学习库:PyTorch(也可以使用TensorFlow平台)

二、安装Anaconda

Anaconda官网:https://www.anaconda.com

点击下载Download即可。Anaconda的下载安装都很简单,网上教程很多。这里就不赘述了。

三、安装Cuda+Cudnn

跟着这个b站博主做的,讲的很通俗易懂,安装一次就成功了。

2022最新版CUDA安装+环境配置,人工智能小白变强教程!!!-人工智能/深度学习/机器学习

先查看自己电脑所支持的CUDA的最高版本是多少。

win+R 输入cmd 进入命令行,输入nvidia-smi,根据右上角的CUDA Version可以知道,该电脑适配12.0版本以下的CUDA。

注意: 安装cuda的版本时不要安装最新版本,很有可能会和之后安装的东西不适配。个人建议是如果和pytorch结合做项目,那么去pytorch官网查看一下目前的版本。个人选择的是CUDA11.7。

CUDA的版本也和Anaconda中下载的python版本有关(这里的python是下载anaconda时自带的,不需要另外去下载)。因为下载anaconda时是直接下载的,所以查看一下python的版本。

win+R 输入cmd 回车,然后输入conda activate激活anaconda环境,输入python -V查看版本

这里查到版本是3.9.0,但其实最开始下载完anaconda后查看版本是3.10,通过查看cuda和python版本的对应关系,发现目前cuda可以支持python的版本只有3.9之前,3.10是不支持的,所以这里需要修改anaconda环境中python的版本,修改为3.9版本。可以根据这个文章里的步骤去修改:https://blog.csdn.net/qq_56520755/article/details/130489115

按照上述的步骤,就可以把Anaconda、cuda和cudnn下载完成。

四、Pytorch(GPU版)

整体的安装环境过程,在这个地方用的时间最长。先列出几个不顺利的方法,再说明我的解决方法。

注意: 如果之后使用过程是在某个虚拟环境中进行时,那么一定要进入自己创建的虚拟环境之后再进行安装。

不顺利的方法:

1、直接在pytorch官网复制conda命令或者pip命令,然后进入Anoconda环境去下载,尝试了5-6次,都没有成功,失败原因是下载时间超过。(这里有个解决办法是可以去设置timeout的时间,但是我没有去尝试,有兴趣的如果碰到这个问题可以试试)

2、因为上一步显示下载时间过长的问题,所以就想到使用国内镜像源去下载,但是这里有个问题是,使用国内镜像源下载(清华镜像源或是其他)下载下来的都是适用于cpu版本,但是并不适用于GPU版本,结果就是卸载重新想办法安装。

成功的办法:离线安装

先在pytorch官网上找到自己CUDA版本对应的pytorch、torchvision应该安装的版本号。寻找网址

再去以下这个地址中去下载自己对应的xml文件: https://download.pytorch.org/whl/torch_stable.html

寻找对应xml文件的方法,网上可以搜到,我底下以自己为例举个例子。

我要下载的是torch11.7,所以我要找到对应 pytorch=2.0.0、torchvision=0.15.0的版本的文件。

conda install pytorch==2.0.0 torchvision==0.15.0 torchaudio==2.0.0 pytorch-cuda=11.7 -c pytorch -c nvidia

我要下载的torch离线包是这个,具体的下载原因是:

cu117其中这个117指的是cuda是11.7版本,torch代表的是torch包,2.0.0是对应我上面找到的torch版本,cp39代表的是python版本是3.9版本,win代表的是windows系统,64代表自己的电脑是64位。

下载的torchvision包是这个,具体的原因与上面相似,这里就不一一赘述。

下载到本地后,先进入Anaconda的环境(创建一个虚拟环境,然后再进入这个虚拟环境,具体的创建和进入的过程网上有很多教程),然后cd进入到自己下载的文件夹的位置,比如我是 cd E:\torch\torch2.0.1,再输入E:回车,再输入pip install 文件名称即可,两个文件都要pip install,记住先pip install torch文件,再pip install 另外一个。

注意: 在安装完成后,在pycharm里运行import torch时报错了,显示的问题是版本不对,所以我又重新下载了pycharm中提示的版本,所以以下例子中是2.0.1,而不是2.0.0。

对以上图片中命令行的解释如下:

首先第一步是进入自己创建的虚拟环境中(我自己创建的虚拟环境叫yolov7,这个是自己取的)

进入环境后进入下载文件的存放位置

再使用pip install 进行下载即可,记得文件要加上文件类型.xml

对于上图为什么之前说的是2.0.0版本,但是我下载的确实2.0.1版本的解释:

根据官网上找到的版本,发现是2.0.0版本,但是当我第一次pip install 完成后,在pycharm里运行import torch报错,根据报错原因里找到了我应该下载的是2.0.1版本,以及对应的torchversion版本是0.15.2,所以我就卸载了之前安装的2.0.0版本,重新安装了提示的版本。这里如何一下子找到正确的版本我没有进行探索,如果有兴趣的可以自己探索安装一下。

查看torch是否安装成功以及是否可以适用GPU:

进入Anaconda虚拟环境,然后输入python,输入import torch导入torch包,再输入torch.__version__查看torch的版本,再通过输入torch.cuda.is_available()结果是True可以判断适用于GPU,如果结果为False,则证明不适用于GPU。

五、安装pycharm

pycharm的安装过程网上教程很多,这里就不介绍了。主要介绍以下pycharm中环境的配置。

在pycharm中打开自己想要运行的文件夹,然后配置解释器。

找到python interpreter后,点击Add interpreter

进入之后点击existing,然后浏览文件找到对应的虚拟环境的python文件。通常虚拟环境保存在下载的anaconda文件夹下的envs文件夹下就可以找到自己创建的所有的虚拟环境,选择自己想要运行的虚拟环境,点击对应文件夹,选择python.exe文件即可。

通过以上的步骤就将解释器设置成功了,然后就可以运行自己的程序了,运行程序时可能会出现没有某个模块的情况,通过pip install下载对应模块即可。

解释器配置完成后,在pycharm中对torch的版本以及对GPU的适用性进行测试,成功则结束。

相关实践学习
基于阿里云DeepGPU实例,用AI画唯美国风少女
本实验基于阿里云DeepGPU实例,使用aiacctorch加速stable-diffusion-webui,用AI画唯美国风少女,可提升性能至高至原性能的2.6倍。
相关文章
|
2天前
|
机器学习/深度学习 PyTorch TensorFlow
Pytorch 与 Tensorflow:深度学习的主要区别(1)
Pytorch 与 Tensorflow:深度学习的主要区别(1)
16 2
|
2天前
|
机器学习/深度学习 PyTorch API
pytorch与深度学习
【5月更文挑战第3天】PyTorch,Facebook开源的深度学习框架,以其动态计算图和灵活API深受青睐。本文深入浅出地介绍PyTorch基础,包括动态计算图、张量和自动微分,通过代码示例演示简单线性回归和卷积神经网络的实现。此外,探讨了模型架构、自定义层、数据加载及预处理等进阶概念,并分享了实战技巧、问题解决方案和学习资源,助力读者快速掌握PyTorch。
33 5
|
2天前
|
机器学习/深度学习 PyTorch 算法框架/工具
【Python机器学习专栏】PyTorch在深度学习中的应用
【4月更文挑战第30天】PyTorch是流行的开源深度学习框架,基于动态计算图,易于使用且灵活。它支持张量操作、自动求导、优化器和神经网络模块,适合快速实验和模型训练。PyTorch的优势在于易用性、灵活性、社区支持和高性能(利用GPU加速)。通过Python示例展示了如何构建和训练神经网络。作为一个强大且不断发展的工具,PyTorch适用于各种深度学习任务。
|
2天前
|
机器学习/深度学习 自然语言处理 算法
PyTorch与NLP:自然语言处理的深度学习实战
随着人工智能技术的快速发展,自然语言处理(NLP)作为其中的重要分支,日益受到人们的关注。PyTorch作为一款强大的深度学习框架,为NLP研究者提供了强大的工具。本文将介绍如何使用PyTorch进行自然语言处理的深度学习实践,包括基础概念、模型搭建、数据处理和实际应用等方面。
|
2天前
|
机器学习/深度学习 并行计算 PyTorch
PyTorch与CUDA:加速深度学习训练
【4月更文挑战第18天】本文介绍了如何使用PyTorch与CUDA加速深度学习训练。CUDA是NVIDIA的并行计算平台,常用于加速深度学习中的矩阵运算。PyTorch与CUDA集成,允许开发者将模型和数据迁移到GPU,利用`.to(device)`方法加速计算。通过批处理、并行化策略及优化技巧,如混合精度训练,可进一步提升训练效率。监控GPU内存和使用调试工具确保训练稳定性。PyTorch与CUDA的结合对深度学习训练的加速作用显著。
|
2天前
|
PyTorch 算法框架/工具 Android开发
PyTorch 深度学习(GPT 重译)(六)(4)
PyTorch 深度学习(GPT 重译)(六)
38 2
|
2天前
|
机器学习/深度学习 PyTorch 算法框架/工具
PyTorch 深度学习(GPT 重译)(六)(3)
PyTorch 深度学习(GPT 重译)(六)
30 2
|
1天前
|
机器学习/深度学习 算法 计算机视觉
深度学习在图像识别中的应用与挑战
【5月更文挑战第16天】 随着科技的不断发展,深度学习技术在图像识别领域取得了显著的成果。本文将探讨深度学习在图像识别中的应用,以及在实际应用中所面临的挑战。我们将介绍卷积神经网络(CNN)在图像识别中的关键作用,以及如何通过优化网络结构和训练策略来提高识别准确率。此外,我们还将讨论在大规模数据集上进行训练时可能遇到的问题,以及如何利用迁移学习和数据增强等技术来解决这些问题。
|
1天前
|
机器学习/深度学习 算法 计算机视觉
深度学习在图像识别中的应用与挑战
【5月更文挑战第16天】 随着人工智能技术的飞速发展,深度学习在图像识别领域取得了显著的成果。本文将探讨深度学习在图像识别中的应用,分析其优势和面临的挑战。我们将重点关注卷积神经网络(CNN)在图像分类、目标检测和语义分割等方面的应用,并讨论数据不平衡、过拟合和计算资源等挑战。最后,我们将展望深度学习在图像识别领域的未来发展趋势。
|
1天前
|
机器学习/深度学习 传感器 自动驾驶
基于深度学习的图像识别技术在自动驾驶系统中的应用
【5月更文挑战第16天】 随着人工智能技术的突飞猛进,特别是深度学习在图像识别领域的应用,已成为推动自动驾驶技术发展的关键因素。本文旨在探讨基于深度学习的图像识别技术如何被集成到自动驾驶系统中,提高车辆的环境感知能力,确保行车安全。我们将分析卷积神经网络(CNN)和循环神经网络(RNN)等深度学习模型在处理实时交通数据中的优势,同时探讨这些技术面临的挑战和潜在的改进方向。通过实验结果验证,基于深度学习的图像识别系统能够有效提升自动驾驶汽车的导航精度与决策效率,为未来智能交通系统的实现奠定基础。
16 4
http://www.vxiaotou.com