一:发现问题
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、CUDA与pytorch版本
CUDA版本:12.1
pytorch版本:1.12.1 cpu
Previous PyTorch Versions | PyTorch