File tree Expand file tree Collapse file tree 2 files changed +7
-7
lines changed Expand file tree Collapse file tree 2 files changed +7
-7
lines changed Original file line number Diff line number Diff line change @@ -14,7 +14,7 @@ export USE_CUDA_STATIC_LINK=1
14
14
. ./switch_cuda_version.sh 9.0
15
15
16
16
17
- conda install -y cmake numpy=1.17 setuptools pyyaml mkl=2018 mkl-include typing_extension ninja magma-cuda80 -c pytorch
17
+ conda install -y cmake numpy=1.17 setuptools pyyaml mkl-static =2018 mkl-include typing_extension ninja magma-cuda80 -c pytorch
18
18
19
19
export CMAKE_PREFIX_PATH=" $( dirname $( which conda) ) /../"
20
20
git clone https://github.com/pytorch/pytorch -b nightly2 --recursive
Original file line number Diff line number Diff line change @@ -21,9 +21,9 @@ requirements:
21
21
- pyyaml
22
22
{% if cross_compile_arm64 == 0 %}
23
23
- mkl-include # [x86_64]
24
- - mkl=2020.2 # [py <= 311 and x86_64 and not win]
25
- - mkl=2023.1 # [py >= 312 and x86_64]
26
- - mkl-devel =2023.1 # [x86_64 and win]
24
+ - mkl-static =2020.2 # [py <= 311 and x86_64 and not win]
25
+ - mkl-static =2023.1 # [py >= 312 and x86_64]
26
+ - mkl-static =2023.1 # [x86_64 and win]
27
27
{% endif %}
28
28
- typing_extensions
29
29
- ninja
@@ -39,8 +39,8 @@ requirements:
39
39
run :
40
40
- python
41
41
{% if cross_compile_arm64 == 0 %}
42
- - mkl >=2018 # [x86_64 and not win]
43
- - mkl=2023.1 # [x86_64 and win]
42
+ - mkl-static >=2018 # [x86_64 and not win]
43
+ - mkl-static =2023.1 # [x86_64 and win]
44
44
{% endif %}
45
45
- libuv # [win]
46
46
- intel-openmp # [win]
@@ -54,7 +54,7 @@ requirements:
54
54
- jinja2
55
55
- pyyaml
56
56
{% if cross_compile_arm64 == 0 %}
57
- - blas * mkl # [x86_64]
57
+ - blas * mkl-static # [x86_64]
58
58
{% endif %}
59
59
- pytorch-mutex 1.0 {{ build_variant }} # [not osx ]
60
60
{{ environ.get('CONDA_CUDATOOLKIT_CONSTRAINT', '') }}
You can’t perform that action at this time.
0 commit comments