本地GPU调用失败问题解决1

一：发现问题

1、电脑环境参数：

OMEN by Gaming Laptop

Windows 11 家庭中文版

2th Gen Intel(R) Core(TM) i9-12900H   2.50 GHz

NVIDIA GeForce RTX 3060 Laptop GPU 显存6G

PyCharm 2023.1.1(Professional Edition)

2、PyCharm中检测GPU：安装版本错误

import torch
print(torch.__version__)
print(torch.cuda.is_available())

返回：

1.12.1

False

cpu

0

3、anaconda验证

输入：

conda env list

conda activate env_pytorch1121

conda list 

返回值中

确定是cpu版本

3、发现最新2.2.0决定重新安装Pytorch版本

具体操作：

1）基于现有anaconda中的环境复制新环境

2）删除其中的pytorch、torchvision和torchaudio

3）安装新的版本，这里准备尝试分别安装最新版pytorch2.2.0和1.12.1（支持cuda11.6）

具体操作过程和效果见下一篇文章

附：其他排查

1、电脑任务管理器显示：正常

2、设备管理器驱动显示：正常

3、检查电脑支持的cuda版本：12.3

cmd输入：nvidia-smi

4、检查CUDA版本：cuda_12.1

cmd输入nvcc --version

与5不同，以此为准，CUDA驱动12.1。

5、检查pytorch版本详细信息

管理员打开anaconda prompt 

输入：conda env list

输入：conda activate env_pythorch1121 当前所使用的1121

6、系统情况

NVIDIA Control Panel

确定CUDA的cudatoolkit的版本：NVIDIA CUDA 12.3.106

7、CUDA与pytorch版本

CUDA版本：12.1 

pytorch版本：1.12.1 cpu

Previous PyTorch Versions | PyTorch