linux环境中安装pytorch

这篇博客介绍了如何在Linux环境下通过conda创建一个专门用于PyTorch的虚拟环境,以避免与其他库冲突。首先,创建一个名为pytorch、Python版本为3.6的环境,然后激活该环境。接着,从PyTorch官网获取安装命令,但需要注意移除末尾的`-c pytorch`选项以使用清华源。配置清华源的步骤包括添加anaconda的main和free频道,并设置显示渠道URL。最后,通过两行命令安装PyTorch和torchvision,并测试GPU可用性。
摘要由CSDN通过智能技术生成
  1. 创建虚拟环境

为pytorch创建一个虚拟环境,这是推荐的做法,这样在创建其他环境如tensorflow时,就不会互相冲突。在终端输入:

conda create -n pytorch python=3.6
这就创建了一个名为pytorch,python版本为3.6的虚拟环境。于是我们可以把Pytorch安装在这个环境下面。首先我们激活这个环境:source activate pytorch
source activate pytorch

Pythorch官网:https://pytorch.org/get-started/previous-versions/

  • 注意:官网上的命令行要把末尾-c pytorch删掉才会用清华源。
    例如:conda install pytorch==1.2.0 torchvision == 0.4.0
    配置清华源:
    https://blog.csdn.net/hktxt/article/details/82111203
    配置:
    conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
    conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
    conda config --set show_channel_urls yes
    查看:
    cat ~/.condarc

这个还不错:
清华镜像原
简单一句话用清华镜像原:真正的安装就两行命令,切换清华镜像原,安装
打开cmd

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值