当前位置：网站首页>One of the experience of solving problems with code error reporting: CUDA does not match pytorch

One of the experience of solving problems with code error reporting: CUDA does not match pytorch

2022-07-28 22:49:00 【Artificial Intelligence Algorithm Research Institute】

ask       topic ：D:\python\envs\yolov5\lib\site-packages\torch\autocast_mode.py:141: UserWarning: User provided device_type of 'cuda', but CUDA is not available. Disabling

  warnings.warn('User provided device_type of \'cuda\', but CUDA is not available. Disabling')

primary       because ：cuda And pytorch Mismatch

import torch

print(torch.__version__)

print(torch.cuda.is_available())

resolvent ： Change autocast_mode.py Comment output statement in

原网站

版权声明
本文为[Artificial Intelligence Algorithm Research Institute]所创，转载请带上原文链接，感谢
https://yzsam.com/2022/196/202207130600561190.html

边栏推荐

猜你喜欢

随机推荐