conda create -n 环境名字 python=3.x cudatoolkit=[10.1/11.1/11.3] cudnn=[7.6/8.x]
注意 cudnn 代表的是算力,要和 cudatoolkit 版本匹配!
一、pip 自动部署工具 requirements.txt
生成当前环境的依赖:
pip freeze > requirements.txt
其中 requirements.txt 文件内容如下。
matplotlib==xxx
Pillow==xxx
使用 requirements.yml 完成环境部署:
conda/source activate 虚拟环境
pip install -r requirements.txt
二、conda 自动部署工具 environment.yml
使用以下命令将当前激活的虚拟环境保存:
conda env export > environment.yml
在当前目录下自动生成相关安装依赖关系,具体示例如下。
name: xxx
channels:
- defaults
- custom
- pytorch
- anaconda
- conda-forge
dependencies:
- python=3.x
- pip
- xxx
- pip:
- xxx
- https://..
例子:
name: smartConstruction
channels:
- defaults
dependencies:
- python=3.8
- pip
- pytorch==1.8.1
- torchvision==0.9.1
- torchaudio==0.8.1
- cudatoolkit=11.3
- pip:
- matplotlib
- opencv-python
- pyyaml
- scipy
- tqdm
- pycocotools
- tensorboard
prefix: /home/guest/anaconda3/envs/smartConstruction
指定安装版本时,注意 pip 使用双等号,dependencies 中为等号!如果不指定,默认安装最新的版本。
使用 envrionment.yml 完成环境部署:
conda env create -f environment.yml
参考链接:https://www.jianshu.com/p/7cc505f8e2bd
|