使用Pytorch安装CuPy

Question

我想把张量从Pytorch传递给CuPy，然后再传递回来做一些Pytorch中没有的操作。将它们安装在一起的推荐方式是什么？

我已经使用conda安装了Pytorch，并且它已经安装了cudatoolkit包。我也更喜欢使用conda来安装cupy。然而，CuPy installation instructions recommend uninstalling cudatoolkit (位于底部)。

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

Answer 1

只需执行conda install -c conda-forge cupy，它就应该可以工作。文档的这一部分已经过时了，而且是to be updated。