conda install pytorch torchvision cudatoolkit=10.2 -c pytorch -c hcc
在安装pytorch 是后面加上-c hcc
make.sh时就可以找到nvcc了
版权声明:本文为u013955915原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
conda install pytorch torchvision cudatoolkit=10.2 -c pytorch -c hcc
在安装pytorch 是后面加上-c hcc
make.sh时就可以找到nvcc了