美文网首页pytorch
CPU深度学习环境配置-PyTorch

CPU深度学习环境配置-PyTorch

作者: dalalaa | 来源:发表于2019-12-27 22:35 被阅读0次

    CPU环境主要用来做深度学习部署,毕竟显存太贵,模型如果能压缩部署在CPU服务器上,能节约大量成本。

    模型部署技术的学习成本比模型训练低得多,下面是我部署CPU开发环境的记录。

    首先从清华源下载最新版的miniconda:

    https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/Miniconda3-latest-Linux-x86_64.sh

    安装:

    bash Miniconda3-latest-Linux-x86_64.sh

    安装完成之后重新打开终端即可使用,接下来按照https://mirror.tuna.tsinghua.edu.cn/help/anaconda/的说明添加conda源(pip源也可以做相应的修改)

    接下来可以创建虚拟环境了,我选择的python版本是3.7。

    conda create -n py37 python=3.7

    切换环境:
    source activate py37

    安装pytorch和torchvision
    conda install pytorch-cpu torchvision

    然后就可以开始干活了。

    相关文章

      网友评论

        本文标题:CPU深度学习环境配置-PyTorch

        本文链接:https://www.haomeiwen.com/subject/qmhjoctx.html