使用Pytorch安装CuPy

Question

我想将张量从Pytorch传递到CuPy，然后再做一些在Pytorch中不可用的操作。建议将它们安装在一起的方式是什么？

我已经使用conda安装了Pytorch，并且已经安装了cudatoolkit软件包。我也想用conda来安装cupy。但是，CuPy安装说明recommend uninstalling cudatoolkit（在最底部）。

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

Answer 1

只需conda install -c conda-forge cupy，它应该可以工作。该部分文档已过时，to be updated。