使用 Pytorch 安装 CuPy
installing CuPy with Pytorch
我想将张量从 Pytorch 传递到 CuPy，然后返回以执行 Pytorch 中不可用的一些操作。将它们安装在一起的推荐方法是什么？
我已经使用conda安装了Pytorch，它已经安装了cudatoolkit包。我也更喜欢使用 conda 来安装 cupy。但是 CuPy 安装说明 recommend uninstalling cudatoolkit（在最底部）。
那么，如何确保 Pytorch 和 CuPy 使用相同的 cuda 版本，以及 CuDNN、NVCC 等？或者我不应该使用 conda 安装 cupy？
只需要 conda install -c conda-forge cupy 就可以了。那部分文档已经过时 to be updated.
我想将张量从 Pytorch 传递到 CuPy，然后返回以执行 Pytorch 中不可用的一些操作。将它们安装在一起的推荐方法是什么？
我已经使用conda安装了Pytorch，它已经安装了cudatoolkit包。我也更喜欢使用 conda 来安装 cupy。但是 CuPy 安装说明 recommend uninstalling cudatoolkit（在最底部）。
那么，如何确保 Pytorch 和 CuPy 使用相同的 cuda 版本，以及 CuDNN、NVCC 等？或者我不应该使用 conda 安装 cupy？
只需要 conda install -c conda-forge cupy 就可以了。那部分文档已经过时 to be updated.