向系统环境变量中添加一项:TORCH_CUDA_ARCH_LIST,值为7.5。 重启pycharm或者cmd,再进行相关操作即可。
上一篇 模型可视化工具------netron
下一篇 Python自学教程第2期-求100以内奇数和
版权所有 (c)2021-2022 MSHXW.COM
ICP备案号:晋ICP备2021003244-6号