Skip to content

Pip和conda配置环境

记录一下自己在配置pytorch环境时遇到的问题。全程使用Ubuntu命令行操作,可以远程操作服务器

env

conda

安装conda

先到miniconda的官网找到对应的下载链接,然后用wget安装。 - 如果下载速度很慢,可以用镜像源, 找到-latest-linux x86_64

wget  https://mirrors.zju.edu.cn/anaconda/miniconda/Miniconda3-latest-Linux-x86_64.sh
bash Miniconda3-latest-Linux-x86_64.sh 
1. 按ENTER浏览用户协议

  1. 之后一路选择yes, 环境会创建到~/miniconda3/envs

  2. 激活conda的设置``` Do you wish to update your shell profile to automatically initialize conda? This will activate conda on startup and change the command prompt when activated. If you'd prefer that conda's base environment not be activated on startup, run the following command when conda is activated:

    conda config --set auto_activate_base false

    You can undo this by running conda init --reverse $SHELL? [yes|no] ```

    出现这个的时候,选择yes 这样启动终端时会自动激活conda

  3. 安装完成后,执行source ~/.bashrc 重启终端,最左边会出现(base)字样,说明安装完成

建立环境

conda create -n [环境名称] python=[版本号]

接着重启终端。然后才能激活环境

conda activate [环境名称]

换源

Info

mirrors.zju.edu.cn目前只支持校园内网访问,外网访问时会自动重定向到清华源

以zju源为例

conda config --add channels https://mirrors.zju.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.zju.edu.cn/anaconda/cloud/conda-forge
conda config --add channels https://mirrors.zju.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.zju.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.zju.edu.cn/anaconda/pkgs/main/

或者直接vim .condarc

结合vscode和conda

参考资料

  1. 用vscode打开一个文件夹
  2. Ctrl+Shift+P打开vsc的命令行
  3. 选择Python:Select Interpreter
  4. vsc会自动定位Python的位置 ~\miniconda3\envs\对应的版本,选择对应版本号和环境名称即可

设置好之后,每次启动vsc集成的终端时就会自动激活对应的环境。python代码也可以点击右上角一键运行了

conda环境打包

打包:

pip install conda-pack
conda pack -n my_env

如果该环境下有-e安装的包,则要使用--ignore-editable-packages,排除-e的包

conda pack -n my_env --ignore-editable-packages

之后会得到my_env.tar.gz文件

重建:

mkdir ~/miniconda3/envs/my_env
tar -xzvf my_env.tar.gz -C ~/miniconda3/envs/my_env
conda env list #查看结果

注意

如果在环境中混用pip和conda安装包,conda打包的时候有部分包会被漏掉,因此重建之后需要手动pip install缺失的包

pip

pip查找包并安装的速度一般比conda快

临时换源:

pip install package_name -i https://mirrors.zju.edu.cn/pypi/web/simple

pip install --upgrade package_name

pytorch安装

Bug

​ 此部分的内容需要更新

conda install pytorch=[版本号]=[具体名字]

注意源自动匹配的版本可能不对。要检查名字后面不是带着_cpu, 而是_cuda11.3_cudnn.... 可以手动指定具体版本号

conda install pytorch=1.10.0=py3.8_cuda11.3_cudnn8.2.0_0
pytorch和cudatoolkit版本对应

torchvision

torch和torchvision版本对应关系

如果源自动匹配的不对,可以离线安装或者手动指定版本: wget https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch/linux-64/torchvision-0.11.1-py38_cu113.tar.bz2

conda install --offline torchvision-0.11.1-py38_cu113.tar.bz2

Comments