深度学习工具网站

文章介绍了Pytorch开发中常用的代码包Apex,以及如何使用Anaconda或Miniconda管理环境。提到了在conda环境中安装软件包的方法,并指出大多数深度学习库可以通过pip安装。此外,还提及了NVIDIA的CUDACloudContainer用于获取深度学习框架相关的Docker镜像。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

【持续更新...】

Pytorch中常见的代码包

  1. Apex (A PyTorch Extension) (真是个好用的东西!搭配pytorch, 1+1>2)

https://nvidia.github.io/apex/index.html

环境配置

  1. 通常,为了方便管理,会使用anaconda进行环境管理,极大的方便了跑baseline面临的环境不一致的问题。但是anaconda比较臃肿,可以使用Miniconda,功能一样(不影响环境安装),反应速度更快。

官网:https://docs.conda.io/en/latest/miniconda.html

示例:在linux中下载相应的.sh安装文件,然后使用bash运行,根据提示一路进行安装即可。最后刷新环境变量。

bash ~/.bashrc
  1. 平时需要安装一些软件包,但是不知道安装的命令行。推荐以下命令,可以找到各个平台安装的命令:

https://command-not-found.com/

  1. 如果想要在anaconda环境中安装一些软件包,但是使用sudo apt-get install命令无法安装的时候,可以直接去anaconda的首页,搜索你想要安装的软件包,会提供很多anaconda环境下安装这些软件包的命令,运行命令即可。以aria2为例:

https://anaconda.org/

  1. 大多数的深度学习包都可以使用pip进行安装。有时候在复现别人代码的时候,需要安装相应版本的包。因此可以去Pypi网站,查找安装包的名称以及相应的发布版本。

https://pypi.org/

  1. 一般,都会使用NVIDIA的显卡,如果要使用docker镜像,可以去CUDA Cloud Container中去寻找并下载相应的镜像。根据相应的深度学习框架(Pytorch或者Tensorflow),查找相应的Tag。下载即可。

https://catalog.ngc.nvidia.com/containers?filters=&orderBy=dateModifiedDESC&query=

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值