AssertionError: CUDA unavailable, invalid device 0 requested

最新推荐文章于 2023-10-09 18:31:30 发布

VIP文章 bashan16045

最新推荐文章于 2023-10-09 18:31:30 发布

阅读量3.4k

点赞数 1

本文链接：https://blog.csdn.net/bashan1604/article/details/118085968

版权

yolov5用gpu训练时报错了，nvcc -V一看，no command

Linux下安装好了cuda之后nvcc -V仍然无法执行

操作一番后，nvcc -V OK, but 问题依旧

How to Install PyTorch with CUDA 10.2

torch.cuda.is_available()返回false的解决办法

进入python查看当前pytorch是否可用gpu，发现是false

>>> import torch
>>> torch.cuda.is_available()
/home/ubuntu/anaconda3/envs/wby_py38/lib/python3.8/site-packages/torch/cuda/__init__.py:52: UserWarning: CUDA initialization: The NVIDIA driver on your system is too old (found version 10010). Please update your GPU driver by downloading and installing a new version from the URL: http://www.nvidia.com/Download/index.aspx Alternatively, go to: https://pytor