admin管理员组

文章数量:1663034

一、问题描述

使用gpu运行深度学习代码时,遇到这个错误。

二、错误原因:

与cuda驱动的版本有关。

三、解决办法:

1.打开命令行输入:nvidia-smi
2.查看CUDA版本

3.打开pytorch官网,下载对应CUDA版本的pytorch

本文标签: GPUnvalidArchArchitecture