使用Pytorch安装CuPy

问题描述 投票:0回答:1

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

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

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

pytorch cudnn cupy
1个回答
0
投票

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

© www.soinside.com 2019 - 2024. All rights reserved.