我想将张量从Pytorch传递到CuPy,然后再传递回来,以执行Pytorch中不可用的一些操作。建议将它们安装在一起的方法是什么?
我已经使用conda安装了Pytorch,它也安装了cudatoolkit包。我更喜欢用conda来安装cupy。但是,CuPy的安装说明建议卸载cudatoolkit(在最底部(。
那么,我如何确保Pytorch和CuPy都使用相同的cuda版本,以及CuDNN、NVCC等?或者我应该不使用conda来安装cupy吗?
只需执行conda install -c conda-forge cupy
,它就会工作。这部分文件已经过时,需要更新。