Skip to content

Instantly share code, notes, and snippets.

@hanxue
Created January 28, 2021 05:48
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save hanxue/2953a044f18ef32bababdfd440bb58a5 to your computer and use it in GitHub Desktop.
Save hanxue/2953a044f18ef32bababdfd440bb58a5 to your computer and use it in GitHub Desktop.
Vitis AI 1.3.0 Caffe GPU conda meta.yaml
package:
name: caffe_decent_gpu
version: {{ environ.get('BUILD_VER', '1.3.0') }}
build:
number: {{ environ.get('BUILD_NUMBER', '1') }}
string: {{ environ["GIT_FULL_HASH"][:8] }}
source:
#path: ../../caffe
git_url: gits@server:repo/caffe.git
git_rev: {{ environ.get('BUILD_BRANCH', 'dev') }}
git_depth: 1
requirements:
build:
- {{ compiler('c') }}
- {{ compiler('cxx') }}
- {{ cdt('mesa-libgl-devel') }} # [linux]
- {{ cdt('mesa-dri-drivers') }} # [linux]
- {{ cdt('libselinux') }} # [linux]
- {{ cdt('libxdamage') }} # [linux]
- {{ cdt('libxxf86vm') }} # [linux]
- {{ cdt('libxext') }} # [linux]
host:
- cudatoolkit {{ cudatoolkit }}
- cudnn {{ cudnn }}
- cmake
- python
- pip
- yaml-cpp
- openblas
- hdf5
- gflags
- glog
- h5py
- ipython
- leveldb
- lmdb
- matplotlib
- networkx
- nose
- numpy
- opencv {{ opencv }}
- pandas
- pillow
- protobuf 3.11.*
- python
- python-dateutil
- python-gflags
- python-leveldb
- pyyaml
- scikit-image
- scipy
- six
- snappy
- boost {{ boost }}
- nccl
- openssl
- xorg-libxfixes # [linux]
- xorg-libxau
run:
- {{ pin_compatible('cudatoolkit', max_pin='x.x') }}
- {{ pin_compatible('cudnn') }}
- yaml-cpp
- openblas
- hdf5
- gflags
- glog
- h5py
- ipython
- leveldb
- lmdb
- matplotlib
- networkx
- numpy
- opencv
- pandas
- pillow
- protobuf 3.11.*
- python
- python-dateutil
- python-gflags
- python-leveldb
- pyyaml
- scikit-image
- scipy
- six
- snappy
- boost
- nccl
- openssl
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment