diff --git a/recipe/bld.bat b/recipe/bld.bat index cc124409..b147d182 100644 --- a/recipe/bld.bat +++ b/recipe/bld.bat @@ -26,12 +26,6 @@ if "%blas_impl%" == "generic" ( SET "USE_NUMA=0" SET "USE_ITT=0" -@REM KINETO seems to require CUPTI and will look quite hard for it. -@REM CUPTI seems to cause trouble when users install a version of -@REM cudatoolkit different than the one specified at compile time. -@REM https://github.com/conda-forge/pytorch-cpu-feedstock/issues/135 -set "USE_KINETO=OFF" - if "%PKG_NAME%" == "pytorch" ( set "PIP_ACTION=install" :: We build libtorch for a specific python version. diff --git a/recipe/meta.yaml b/recipe/meta.yaml index 91f0ee79..b41b19e7 100644 --- a/recipe/meta.yaml +++ b/recipe/meta.yaml @@ -126,14 +126,14 @@ requirements: {% if cuda_compiler_version != "None" %} - cuda-driver-dev # [linux] - cuda-cudart-dev - - cuda-cupti-dev # [linux] + - cuda-cupti-dev - cuda-nvrtc-dev - cuda-nvtx-dev - cuda-nvml-dev - cuda-profiler-api - - cusparselt # [linux] + - cusparselt - libcublas-dev - - libcudss-dev # [linux] + - libcudss-dev - libcufile-dev # [linux] - libcufft-dev - libcurand-dev @@ -266,14 +266,14 @@ outputs: {% if cuda_compiler_version != "None" %} - cuda-driver-dev # [linux] - cuda-cudart-dev - - cuda-cupti-dev # [linux] + - cuda-cupti-dev - cuda-nvrtc-dev - cuda-nvtx-dev - cuda-nvml-dev - cuda-profiler-api - - cusparselt # [linux] + - cusparselt - libcublas-dev - - libcudss-dev # [linux] + - libcudss-dev - libcufile-dev # [linux] - libcufft-dev - libcurand-dev