deeplabcut cpu版本安装

最新推荐文章于 2023-08-14 11:27:29 发布

vegetable_bird1

最新推荐文章于 2023-08-14 11:27:29 发布

阅读量968

点赞数 1

文章标签： cuda python tensorflow linux anaconda

本文链接：https://blog.csdn.net/vegetable_bird1/article/details/118895375

版权

deeplabcut CPU版本安装教程

在win10下：

法一，新建环境在安装

1在annaconda中新建虚拟环境

conda create -n <name_of_the_environment> python=3.6

2进入虚拟环境

 activate <name_of_the_environment>

3安装deeplabcut

pip install deeplabcut

4安装图形化界面

pip install -U wxPython

5安装TensorFlow CPU版本

pip install tensorflow==1.10

6若以上步骤numpy版本报错，

pip install numpy==1.17.3

如果显卡，需要安装CUDA11、tf2.5的版本。目前推荐的列表为：

    tensorflow-gpu=2.5

    cudatoolkit=11.3.1

    cudnn=8.2.1

    deeplabcut=2.2rc3

7验证安装是否成功

ipython
in[1]:import deeplabcut

若不报错，则安装完成。
或通过testscript.py验证

法二

通过annaconda图形界面导入yaml文件安装。

法三

下载官方的配置文文件，链接
在conda中将目录切换到配置文件所在目录后，运行

conda env create -f DEEPLABCUT.yaml

若报错，可尝试注释掉，channel字段。

name: DEEPLABCUT
# channels:
#   - conda-forge
#   - defaults
dependencies:
  - python=3.8
  - pip
  - jupyter
  - nb_conda
  - ffmpeg
  - pip:
    - "deeplabcut[gui]==2.2rc3"

在linux下：

通过配置文件安装，若操作系统无gui，则不能安装gui版本，需改为如下，否则安装报错：

name: DEEPLABCUT
# channels:
#   - conda-forge
#   - defaults
dependencies:
  - python=3.8
  - pip
  - jupyter
  - nb_conda
  - ffmpeg
  - pip:
    - "deeplabcut==2.2rc3"

notes：

若用自带的测试脚本测试报错，无法下载预训练ResNets时，可手动到提示网站下载压缩包，解压后的resnet_v1_50.ckpt放到下面目录：

/opt/anaconda3/envs/DEEPLABCUT/lib/python3.8/site-packages/deeplabcut/pose_estimation_tensorflow/models/pretrained