Apex: Conda package

Created on 31 Jul 2019  路  9Comments  路  Source: NVIDIA/apex

Is there a conda package for this or can I only install it via pip?

Most helpful comment

with pip install, I cannot use cudatoolkit from conda (I don't have nvcc and seperated CUDA installation because it is conflict with versioning, and so much bug, trouble going)

So, I am urgent to have CUDA for full feature apex support with CUDA

All 9 comments

We are currently in the process of releasing a conda package.
The pip package is not related to NVIDIA/apex, but refers to this package.

Oh! My bad, I thought it's the pip package since your installation steps mention pip.

Any chance that you know a rough release date of the conda package?

+1 on the conda package, would make things much easier.

+1, really need conda package to solve the CUDA dependency mess

with pip install, I cannot use cudatoolkit from conda (I don't have nvcc and seperated CUDA installation because it is conflict with versioning, and so much bug, trouble going)

So, I am urgent to have CUDA for full feature apex support with CUDA

Any update on this? There are numerous issues related to building apex which could easily be avoided by a conda package that tightly couples:

pytorch <--> apex <--> cudatoolkit

This would make life a whole lot easier for a lot of folks.

I would also be interested to know about any updates.

Pytorch 1.6 will have native mixed precision training, and I assume the native sync-batch-norm will be supported.

We decided to focus on getting the most important utilities from apex into upstream, so no separate installs are needed. Pytorch 1.6 will have native mixed precision support with a better API.

Was this page helpful?
0 / 5 - 0 ratings