一、查看显卡是否配置正确
驱动下载地址:https://www.nvidia.cn/geforce/geforce-experience/
二、chuda和cudnn设置
cuda下载地址:https://developer.nvidia.com/cuda-11-8-0-download-archive
cudnn下载地址:https://developer.nvidia.com/rdp/cudnn-archive
查看cuda :nvcc –version
查看pytorch版本:在项目下面输入python.exe
import torch
print(torch.__version__)
三、电脑性能查看
开启卓越模式:powercfg -duplicatescheme e9a42b02-d5df-448d-aa00-03f14749eb61
四、查看cuda是否正常工作
命令窗口查看输入:nvidia-smi
五、显存不够内存来凑,强制开启 选择内存回退
原文链接:https://ai.huanxian.art:33/archives/1409,转载请注明出处。
评论3