1.6.0版本的torch不支持低版本的cuda 10.1 所以检测不到gpu，安装兼容10.1低版本cuda的对应torch版本1.6.0+cu101
pip install torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html
免责声明！
本站转载的文章为个人学习借鉴使用，本站对版权不负任何法律责任。如果侵犯了您的隐私权益，请联系本站邮箱yoyou2525@163.com删除。
猜您在找
RuntimeError: CUDA error: invalid device ordinal
Ubuntu unavailable modifier requested: 0
RuntimeError: cuda runtime error (10) : invalid device ordinal
指定用第几块GPU、CUDA error: invalid device ordinal、
AssertionError: Torch not compiled with CUDA enabled
TensorFlow学习笔记 速记2 报错：failed call to cuDevicePrimaryCtxRetain: CUDA_ERROR_INVALID_DEVICE
CUDA ---- device管理
uwsgi -- unavailable modifier requested: 0 -- 的解决办法
AssertionError
AssertionError