Pytorch环境配置

文章讨论了在深度强化学习(DRL)中,由于神经网络维度较小,CPU的运算速度可能超过GPU。尽管GPU通常因其大规模并行运算能力而闻名,但在这种特定场景下,其优势无法充分发挥,数据传输开销反而成为性能瓶颈,导致速度较CPU慢。
摘要由CSDN通过智能技术生成

大数据中心:

module load anaconda3

conda create --name XXX python=XXX    # 创建虚拟环境

source activate XXX    # 激活虚拟环境

在激活虚拟环境之后,可以使用conda activate XXX和conda deactivate进入环境或者退出环境

Pytorch安装:进入pytorch.org找到Conda安装对应的命令

例如cpu版本:conda install pytorch==XXX torchvision torchaudio cpuonly -c pytorch

实验室集群:离线形式安装,Anaconda安装包在/home/publicuser下

注:DRL神经网络维度较小,导致CPU运算速度要远快于GPU。GPU擅长的是大规模并行运算,相较CPU的优势在于拥有巨量核心数,但单核速度远远低于CPU,同时数据传输开销远大于CPU。在DRL网络维度较小的场景下,GPU核心数多的优势发挥不出来,数据传输开销大的不足表现明显,因此速度反而慢于CPU。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值