使用Pytorch安装CuPy



我想将张量从Pytorch传递到CuPy,然后再传递回来,以执行Pytorch中不可用的一些操作。建议将它们安装在一起的方法是什么?

我已经使用conda安装了Pytorch,它也安装了cudatoolkit包。我更喜欢用conda来安装cupy。但是,CuPy的安装说明建议卸载cudatoolkit(在最底部(。

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

只需执行conda install -c conda-forge cupy,它就会工作。这部分文件已经过时,需要更新。

最新更新