【2020华为云AI实战营】Modelarts Notebook中搭建python3.7+pytorch1.5环境 (一)
Notebook中搭建python3.7+pytorch1.5环境
一、背景及准备工作
(1)背景
目前ModelArts最新推出了Python3.6+Tensorflow 2.1.0环境,CUDA版本10.1,基于这个环境可以搭建最近的很多新版本的环境,但是没有Python3.7的环境,比如想复现一些模型对环境有要求,例如yoloV5需要python3.7+pytorch1.5,基于这种情况在Notebook中怎么进行环境搭建呢?
(2)准备工作
进入Modelarts控制台,新建Notebook,选择配置如下
二、体验JupyterLab
环境启动后进入Lab,jupyterLab官方教程:https://jupyterlab.readthedocs.io/en/stable/,如下图点击Terminal,进入Terminal
三、环境搭建
这个环境搭建就是大家日常的环境搭建流程,咱们再走一遍吧:
(一)基本环境Python3.7
(1)查看环境:conda info --env
(2)新建环境:conda create -n py37 python=3.7
(3) 激活环境:activate py37
(4)删除环境(这里用不到,Notebook停止后,容器自动删除):conda remove -n py37 --all
(5)退出环境:deactivate
(二)深度学习框架Pytorch1.5安装
(1)激活环境:activate py37 后输入:pip install -r yolov5/requirements.txt
requirements.txt内容如下:
# pip install -U -r requirements.txt
Cython
numpy==1.17
opencv-python
torch>=1.4
matplotlib
pillow
tensorboard
PyYAML>=5.3
torchvision
scipy
tqdm
git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI
# Nvidia Apex (optional) for mixed precision training --------------------------
# git clone https://github.com/NVIDIA/apex && cd apex && pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" . --user && cd .. && rm -rf apex
# Conda commands (in place of pip) ---------------------------------------------
# conda update -yn base -c defaults conda
# conda install -yc anaconda numpy opencv matplotlib tqdm pillow ipython
# conda install -yc conda-forge scikit-image pycocotools tensorboard
# conda install -yc spyder-ide spyder-line-profiler
# conda install -yc pytorch pytorch torchvision
# conda install -yc conda-forge protobuf numpy && pip install onnx==1.6.0 # https://github.com/onnx/onnx#linux-and-macos
如果出现下面报错,可以先安装numpy后再执行
pip install -r yolov5/requirements.txt
(2)环境安装好后可以查看一下扩展库是否完整:pip list 其他 缺什么库就装什么库~~~
四、其他环境类似配置安装
已经购买专属资源包年的企业或者开发者可以将Notebook环境配置好后,将Notebook一直启动,使用配置的环境。
以上。
- 点赞
- 收藏
- 关注作者
评论(0)