今天发现 torch.cuda.is_available()==false 无法初始化GPU进行训练.
于是着手开始检查torch的版本和cuda的版本
检查torch的版本
输入python
>>> import torch
>>> print(torch.__version__)
如果带有cpu字样说明你装的不是 gpu版本的, 需要重新安装pytorch
我的是cpu版本的. 于是重装.
再次测试
输入python
>>> import torch
>>> print(torch

博主在使用PyTorch时遇到GPU不可用的问题,通过检查发现安装的是CPU版本而非GPU版本。经过重新安装并对比cuda和nvidia-smi的版本,发现CUDA驱动为10.2,解决方法是下载并安装最新官方驱动。
最低0.47元/天 解锁文章
1855





