前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >搭建深度学习环境(Pytorch)

搭建深度学习环境(Pytorch)

作者头像
算法与编程之美
发布2023-01-03 19:29:07
5940
发布2023-01-03 19:29:07
举报
文章被收录于专栏:算法与编程之美

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. .设置相关环境,网上有很多教程。
  1. .检测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文件的导入模块名,就可以使用了。

本文参与 腾讯云自媒体同步曝光计划,分享自微信公众号。
原始发表:2022-12-03,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 算法与编程之美 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
相关产品与服务
GPU 云服务器
GPU 云服务器(Cloud GPU Service,GPU)是提供 GPU 算力的弹性计算服务,具有超强的并行计算能力,作为 IaaS 层的尖兵利器,服务于生成式AI,自动驾驶,深度学习训练、科学计算、图形图像处理、视频编解码等场景。腾讯云随时提供触手可得的算力,有效缓解您的计算压力,提升业务效率与竞争力。
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档