一、环境配置
在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