跑深度学习代码时,发现不是在GPU上跑
发现我下载的pytorch是cpu版本的
那还能怎么办,只能卸载了重新来…
首先,查看自己电脑兼容的最大cuda版本
输入nvidia-smi指令
如红色标记,,本机配置最大版本可以安装到11.4,所以CUDA version11.4以下都可以安装。
本次决定安装 version 11.1
后面就参考这篇文章
cuda装成CPU版本如何装回来
于 2022-09-12 10:34:13 首次发布
跑深度学习代码时,发现不是在GPU上跑
发现我下载的pytorch是cpu版本的
那还能怎么办,只能卸载了重新来…
首先,查看自己电脑兼容的最大cuda版本
输入nvidia-smi指令
如红色标记,,本机配置最大版本可以安装到11.4,所以CUDA version11.4以下都可以安装。
本次决定安装 version 11.1
后面就参考这篇文章