Мне кажется, что проблема заключается в номере версии. Во-первых, чтобы указать конкретную версию, вы не можете использовать знак '='. Если вы особенно ищете ту же версию, которую вы указали, я бы рекомендовал попробовать pytorch0.4.1 в качестве имени пакета, иначе вы можете получить последнюю версию с помощью:
conda install -c pytorch pytorch
Вы можете обратиться к этому сайту для CUDA: https://anaconda.org/anaconda/cudatoolkit
или, если это вам подходит, запустите эту команду
pip3 install pycuda