pytorch环境管理

2020-09-12  本文已影响0人  顾子豪

添加国内镜像

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/bioconda
conda config --set show_channel_urls yes

查看国内原:

conda config --show-sources

1创建pytorch虚拟环境:

conda create --name pytorch python=3.7

2进入虚拟环境:

conda activate pytorch

3安装CPU版本的pytorch:

conda install pytorch-cpu torchvision-cpu -c pytorch

4新建的虚拟环境没有jupyter,安装一下jupyter

conda install jupyter

5查看pytorch是否安装成功

import torch
import torchvision

torch.__version__

6关闭虚拟环境

conda deactivate

7删除虚拟环境

conda remove -n name --all

8在环境已激活的情况下使用conda导出已有环境

conda env export>environment.yaml

9使用移到其他服务器上的yaml文件

conda env create -f environment.yaml
上一篇下一篇

猜你喜欢

热点阅读