搭建深度学习环境(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文件的导入模块名,就可以使用了。

目录
相关文章
|
13天前
|
机器学习/深度学习 数据采集 TensorFlow
使用Python实现深度学习模型:智能垃圾分类与环境保护
使用Python实现深度学习模型:智能垃圾分类与环境保护 【8月更文挑战第7天】
28 2
|
2天前
|
机器学习/深度学习 人工智能 PyTorch
【深度学习】使用PyTorch构建神经网络:深度学习实战指南
PyTorch是一个开源的Python机器学习库,特别专注于深度学习领域。它由Facebook的AI研究团队开发并维护,因其灵活的架构、动态计算图以及在科研和工业界的广泛支持而受到青睐。PyTorch提供了强大的GPU加速能力,使得在处理大规模数据集和复杂模型时效率极高。
112 59
|
4天前
|
机器学习/深度学习 自然语言处理 数据处理
|
6天前
|
机器学习/深度学习 PyTorch TensorFlow
【PyTorch】PyTorch深度学习框架实战(一):实现你的第一个DNN网络
【PyTorch】PyTorch深度学习框架实战(一):实现你的第一个DNN网络
29 1
|
8天前
|
机器学习/深度学习 数据采集 TensorFlow
使用Python实现深度学习模型:智能环境监测与预警
【8月更文挑战第11天】 使用Python实现深度学习模型:智能环境监测与预警
29 2
|
17天前
|
机器学习/深度学习 人工智能 PyTorch
【Deepin 20深度探索】一键解锁Linux深度学习潜能:从零开始安装Pytorch,驾驭AI未来从Deepin出发!
【8月更文挑战第2天】随着人工智能的迅猛发展,深度学习框架Pytorch已成为科研与工业界的必备工具。Deepin 20作为优秀的国产Linux发行版,凭借其流畅的用户体验和丰富的软件生态,为深度学习爱好者提供理想开发平台。本文引导您在Deepin 20上安装Pytorch,享受Linux下的深度学习之旅。
39 12
|
13天前
|
机器学习/深度学习 存储 PyTorch
【深度学习】Pytorch面试题:什么是 PyTorch?PyTorch 的基本要素是什么?Conv1d、Conv2d 和 Conv3d 有什么区别?
关于PyTorch面试题的总结,包括PyTorch的定义、基本要素、张量概念、抽象级别、张量与矩阵的区别、不同损失函数的作用以及Conv1d、Conv2d和Conv3d的区别和反向传播的解释。
36 2
|
13天前
|
机器学习/深度学习 算法 PyTorch
【深度学习】TensorFlow面试题:什么是TensorFlow?你对张量了解多少?TensorFlow有什么优势?TensorFlow比PyTorch有什么不同?该如何选择?
关于TensorFlow面试题的总结,涵盖了TensorFlow的基本概念、张量的理解、TensorFlow的优势、数据加载方式、算法通用步骤、过拟合解决方法,以及TensorFlow与PyTorch的区别和选择建议。
33 2
|
20天前
|
机器学习/深度学习 数据挖掘 TensorFlow
解锁Python数据分析新技能,TensorFlow&PyTorch双引擎驱动深度学习实战盛宴
【7月更文挑战第31天】在数据驱动时代,Python凭借其简洁性与强大的库支持,成为数据分析与机器学习的首选语言。**数据分析基础**从Pandas和NumPy开始,Pandas简化了数据处理和清洗,NumPy支持高效的数学运算。例如,加载并清洗CSV数据、计算总销售额等。
33 2
|
20天前
|
机器学习/深度学习 数据挖掘 TensorFlow

热门文章

最新文章