搭建深度学习环境(Pytorch)

简介: 搭建深度学习环境(Pytorch)

1 配置问题

(1).针对于电脑中配备有GPU,且有深度学习需求,搭建一个可用无污染的深度学习环境。

(2).按照Anaconda3 + CUDA10.0 + CuDNN + Pytorch1.2 + Pycharm配置。

(3).解决配置Pytorch中无法使用torchvision库的问题。

2 搭建过程

(1).安装Anaconda。按照自己电脑的需求下载,下载完成后,解压安装包,按照指示一步一步完成即可,检验安装是否成功,并创建一个虚拟环境用于之后的配置。

# 创建虚拟环境:(在cmd终端输入)
conda create -n Pytorch_GPU python=3.7

(2).安装CUDA10.0和cuDNN(必须与自己的Pytorch==1.2.0版本对应起来)

  1. .在官网找到合适的版本进行下载并进行安装。
  2. .找到安装的位置对cuDNN进行解压。(3).安装Pytorch。
  1. . 激活并进入Anaconda提前准备好的虚拟环境。

activate Pytorch_GPU

2).下载安装Pytorch。打开pytorch的官网,地址:https://pytorch.org/

也可以去下载轮子(https://download.pytorch.org/whl/torch_stable.html

(4)Pycharm和Anaconda连接。

  1. .设置相关环境,网上有很多教程。
  2. .检测Pytorch配置成功。

    (5).torchvision问题。
    在导入torchvision的时候出现了报错,但是查询发现torchvision包在库里。
    在网上百度找到问题的解决办法。

    找到报错路径C:\Users\Public\Anaconda3\Lib\site-packages\PIL,打开__init__.py文件,使用from PIL import Image, ImageOps, ImageEnhance, __version__ 替换文件中from PIL import Image, ImageOps, ImageEnhance,PILLOW_VERSION这句。
  3. 结语

1.要注意各个软件版本之间的匹配问题。

2.最新版本的CUDA不再需要配置Cudnn进行加速了。

3.针对torchvision导入模块缺失问题,更改__init__.py文件的导入模块名,就可以使用了。

目录
相关文章
|
8月前
|
机器学习/深度学习 人工智能 PyTorch
PyTorch深度学习 ? 带你从入门到精通!!!
🌟 蒋星熠Jaxonic,深度学习探索者。三年深耕PyTorch,从基础到部署,分享模型构建、GPU加速、TorchScript优化及PyTorch 2.0新特性,助力AI开发者高效进阶。
PyTorch深度学习 ? 带你从入门到精通!!!
|
9月前
|
机器学习/深度学习 存储 PyTorch
Neural ODE原理与PyTorch实现:深度学习模型的自适应深度调节
Neural ODE将神经网络与微分方程结合,用连续思维建模数据演化,突破传统离散层的限制,实现自适应深度与高效连续学习。
804 3
Neural ODE原理与PyTorch实现:深度学习模型的自适应深度调节
|
机器学习/深度学习 数据可视化 算法
PyTorch生态系统中的连续深度学习:使用Torchdyn实现连续时间神经网络
神经常微分方程(Neural ODEs)是深度学习领域的创新模型,将神经网络的离散变换扩展为连续时间动力系统。本文基于Torchdyn库介绍Neural ODE的实现与训练方法,涵盖数据集构建、模型构建、基于PyTorch Lightning的训练及实验结果可视化等内容。Torchdyn支持多种数值求解算法和高级特性,适用于生成模型、时间序列分析等领域。
746 77
PyTorch生态系统中的连续深度学习:使用Torchdyn实现连续时间神经网络
|
12月前
|
机器学习/深度学习 存储 PyTorch
PyTorch + MLFlow 实战:从零构建可追踪的深度学习模型训练系统
本文通过使用 Kaggle 数据集训练情感分析模型的实例,详细演示了如何将 PyTorch 与 MLFlow 进行深度集成,实现完整的实验跟踪、模型记录和结果可复现性管理。文章将系统性地介绍训练代码的核心组件,展示指标和工件的记录方法,并提供 MLFlow UI 的详细界面截图。
524 2
PyTorch + MLFlow 实战:从零构建可追踪的深度学习模型训练系统
|
机器学习/深度学习 PyTorch TensorFlow
深度学习工具和框架详细指南:PyTorch、TensorFlow、Keras
在深度学习的世界中,PyTorch、TensorFlow和Keras是最受欢迎的工具和框架,它们为研究者和开发者提供了强大且易于使用的接口。在本文中,我们将深入探索这三个框架,涵盖如何用它们实现经典深度学习模型,并通过代码实例详细讲解这些工具的使用方法。
1296 0
|
机器学习/深度学习 算法 PyTorch
深度学习笔记(十三):IOU、GIOU、DIOU、CIOU、EIOU、Focal EIOU、alpha IOU、SIOU、WIOU损失函数分析及Pytorch实现
这篇文章详细介绍了多种用于目标检测任务中的边界框回归损失函数,包括IOU、GIOU、DIOU、CIOU、EIOU、Focal EIOU、alpha IOU、SIOU和WIOU,并提供了它们的Pytorch实现代码。
5346 1
深度学习笔记(十三):IOU、GIOU、DIOU、CIOU、EIOU、Focal EIOU、alpha IOU、SIOU、WIOU损失函数分析及Pytorch实现
|
机器学习/深度学习 PyTorch 调度
在Pytorch中为不同层设置不同学习率来提升性能,优化深度学习模型
在深度学习中,学习率作为关键超参数对模型收敛速度和性能至关重要。传统方法采用统一学习率,但研究表明为不同层设置差异化学习率能显著提升性能。本文探讨了这一策略的理论基础及PyTorch实现方法,包括模型定义、参数分组、优化器配置及训练流程。通过示例展示了如何为ResNet18设置不同层的学习率,并介绍了渐进式解冻和层适应学习率等高级技巧,帮助研究者更好地优化模型训练。
1143 4
在Pytorch中为不同层设置不同学习率来提升性能,优化深度学习模型
|
机器学习/深度学习 监控 PyTorch
深度学习工程实践:PyTorch Lightning与Ignite框架的技术特性对比分析
在深度学习框架的选择上,PyTorch Lightning和Ignite代表了两种不同的技术路线。本文将从技术实现的角度,深入分析这两个框架在实际应用中的差异,为开发者提供客观的技术参考。
527 7
|
机器学习/深度学习 数据挖掘 数据处理
深度学习之卫星图像中的环境监测
基于深度学习的卫星图像环境监测是指通过使用深度学习模型处理和分析来自卫星的遥感数据,以实现对地球环境的自动化监测和分析。这项技术极大提升了环境监测的效率、精度和规模,应用于气候变化研究、生态保护、自然灾害监测、城市扩张评估等多个领域。
650 1

热门文章

最新文章