GPU
文章平均质量分 91
在屏幕前出油
学不完的技术,出不完的油
展开
-
显卡版本(算力)、驱动版本(Driver Version)、CUDA Toolkit(CUDA Version)、PyTorch版本之间的匹配
显卡版本(算力)、驱动版本(Driver Version)、CUDA Toolkit(CUDA Version)和PyTorch版本这四者之间有一定的匹配关系。简单地说就是:需要按照显卡型号(算力)安装驱动,再根据驱动版本选择相应版本的CUDA Toolkit进行安装(CUDA Toolkit可安装多个版本)。原创 2023-03-31 21:00:14 · 12393 阅读 · 6 评论 -
关于torch.cuda.is_available()返回False
一、检查GPU型号;二、检查GPU是否支持CUDA;三、检查驱动版本(Driver Version)是否和CUDA Toolkit版本(CUDA Version)匹配原创 2023-12-04 11:51:50 · 1531 阅读 · 0 评论