File tree Expand file tree Collapse file tree 2 files changed +2
-8
lines changed
Expand file tree Collapse file tree 2 files changed +2
-8
lines changed Original file line number Diff line number Diff line change 11{% set version = "0.8.dev" %}
22
33package :
4- name : caffe2-cuda%%CUDA_VERSION%%-cudnn%%CUDNN_VERSION%%
4+ name : caffe2-cuda
55 version : {{ version }}
66
77source :
Original file line number Diff line number Diff line change @@ -31,16 +31,10 @@ if [[ "${BUILD_ENVIRONMENT}" == *cuda* ]]; then
3131 # the package name in meta.yaml based off of these values, we let Caffe2
3232 # take the CUDA and cuDNN versions that it finds in the build environment,
3333 # and manually set the package name ourself.
34- # NOTE: These are magic strings that exist in the meta.yaml
3534 # WARNING: This does not work on mac.
36- sed -i " s/%%CUDA_VERSION%%/${CAFFE2_CUDA_VERSION} /" " ${CAFFE2_ROOT} /conda/cuda/meta.yaml"
37- sed -i " s/%%CUDNN_VERSION%%/${CAFFE2_CUDNN_VERSION} /" " ${CAFFE2_ROOT} /conda/cuda/meta.yaml"
35+ sed -i " s/caffe2-cuda/caffe2-cuda${CAFFE2_CUDA_VERSION} -cudnn${CAFFE2_CUDNN_VERSION} /" " ${CAFFE2_ROOT} /conda/cuda/meta.yaml"
3836
3937 conda build " ${CAFFE2_ROOT} /conda/cuda" ${CONDA_BLD_ARGS[@]} " $@ "
40-
41- # Change the names back
42- sed -i " s/${CAFFE2_CUDA_VERSION} /%%CUDA_VERSION%%/" " ${CAFFE2_ROOT} /conda/cuda/meta.yaml"
43- sed -i " s/${CAFFE2_CUDNN_VERSION} /%%CUDNN_VERSION%%/" " ${CAFFE2_ROOT} /conda/cuda/meta.yaml"
4438else
4539 conda build " ${CAFFE2_ROOT} /conda/no_cuda" ${CONDA_BLD_ARGS[@]} " $@ "
4640fi
You can’t perform that action at this time.
0 commit comments