NVIDIA-SMI 430.40 Driver Version:430.40 CUDA Version:10.1
-----------------------------解决方法1-该方法没有解决问题----------------------------------------
cuda版本与pytorch 版本不一致
查看cuda版本代码:
查看pytorch版本代码:
查看cuda和pytorch 版本对应关系-改为cuda10.1
--------------------------------------解决方法2-问题解决!!!-----------------------------------------
light-the-torch 封装旨在解决此类问题。
pip install light-the-torch
ltt install torch torchvision
ltt install 是 pip install 的直接替代品,无需担心计算后端:
$ ltt install torch torchvision
[...]
Successfully installed future-0.18.2 numpy-1.19.0 pillow-7.2.0 torch-1.5.1+cu101 torchvision-0.6.1+cu101
[...]
ltt install
is also able to handle packages that depend on PyTorch distributions: