在百度飞桨平台上使用pytorch框架

一、环境配置

        在notebooke上直接pip安装pytorch,服务器会因为兼容性的原因制止本次安装,因此需要新建一个终端,在终端中新建虚拟环境—>安装pytorch

conda init         #初始化虚拟环境,执行该命令后要重启一次终端
conda env list     #查看当前的虚拟环境(可以略过)

# 创建虚拟环境并进入
conda create -n 虚拟环境名称 python=版本号 pip 

 #(推荐带上pip,否则有可能创建的虚拟环境没有解释器)

# 进入虚拟环境
conda activate 创建的虚拟环境的名称

# 安装pytorch
# 在官网上检索,按需复制相应命令执行即可



可以换清华源加快速度,网络上有很多教程,在此不再赘述。

二、脚本

        在百度飞桨配置pytorch框架有一个问题在于,每次重启项目时,之前配置好的环境会被删除,需要重新配置。

        因此可以将之前使用的命令保存进一个脚本文件env.sh中,之后每次重新进入项目时运行该脚本即可自动配置文件,配置的环境在服务器自带的python35-paddle120-env。

        

sh env.sh # 运行脚本文件,根据实际情况更改为env文件的实际地址

env.sh文件的参考内容如下:

#!/bin/sh
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --set show_channel_urls yes

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torch==1.1.0 torchvision==0.3.0

  • 5
    点赞
  • 16
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值