Metapackage to select the PyTorch variant. Use conda's pinning mechanism in your environment to control which variant you want.
conda install atalman::pytorch-cuda