File tree 2 files changed +2
-8
lines changed
2 files changed +2
-8
lines changed Original file line number Diff line number Diff line change 1
1
{% set version = "0.8.dev" %}
2
2
3
3
package :
4
- name : caffe2-cuda%%CUDA_VERSION%%-cudnn%%CUDNN_VERSION%%
4
+ name : caffe2-cuda
5
5
version : {{ version }}
6
6
7
7
source :
Original file line number Diff line number Diff line change @@ -31,16 +31,10 @@ if [[ "${BUILD_ENVIRONMENT}" == *cuda* ]]; then
31
31
# the package name in meta.yaml based off of these values, we let Caffe2
32
32
# take the CUDA and cuDNN versions that it finds in the build environment,
33
33
# and manually set the package name ourself.
34
- # NOTE: These are magic strings that exist in the meta.yaml
35
34
# 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"
38
36
39
37
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"
44
38
else
45
39
conda build " ${CAFFE2_ROOT} /conda/no_cuda" ${CONDA_BLD_ARGS[@]} " $@ "
46
40
fi
You can’t perform that action at this time.
0 commit comments