用Pytorch安装CuPy

我想把张量从Pytork传递到CuPy，再传递回来，做一些Pytork中没有的运算。建议用什么方法将它们安装在一起

我已经使用conda安装了Pytorch，它已经安装了cudatoolkit包。我更喜欢用康达来安装丘比以及。但是，CuPy安装说明（在最底部）

---

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

只要执行

conda安装-c conda forge cupy即可，它应该可以工作。该部分文档已经过时。
只是为了澄清：康达安装的CuPy会使用系统范围的CUDA安装，还是康达安装的cudatoolkit（由康达安装的Pytorch安装）？另外，我看到了关于nvcc的说明，所以我需要在安装CuPy后运行
conda install nvcc
和
conda install cudnn
？噢，嗨，Michael，直到现在才看到你的评论。我想是吧？