使用 Pytorch 安装 CuPy

【问题标题】：installing CuPy with Pytorch使用 Pytorch 安装 CuPy
【发布时间】：2020-06-17 10:03:10
【问题描述】：

我想将张量从 Pytorch 传递给 CuPy，然后返回执行 Pytorch 中不可用的一些操作。将它们安装在一起的推荐方式是什么？

我已经使用 conda 安装了 Pytorch，它已经安装了 cudatoolkit 包。我也更喜欢使用 conda 来安装 cupy。然而，CuPy 安装说明recommend uninstalling cudatoolkit（在最底部）。

那么，我如何确保 Pytorch 和 CuPy 使用相同的 cuda 版本，以及 CuDNN、NVCC 等？还是我不应该使用 conda 来安装 cupy？

【问题讨论】：

【解决方案1】：

只需执行conda install -c conda-forge cupy 即可。那部分文档已经过时了，to be updated.

【讨论】：

澄清一下：安装 conda 的 CuPy 将使用系统范围的 CUDA 安装，还是安装 conda 的 cudatoolkit（由 conda 安装的 Pytorch 安装）？另外，我看到了关于 nvcc 的注释，所以我需要在安装 CuPy 后运行conda install nvcc 和conda install cudnn 吗？
哦，嗨，迈克尔，直到现在才看到您的评论。我想I answered your questions?