win10 torchvision训练模型错误()

pytorch-1.8.2+cuda10.2+torchvison-0.9.2(有问题)

pytorch/vision链接

1、拉vision代码

1)代码版本号需要跟pytorch对应上,也可以跟torchvison版本一致

2、重新编译生成torchvision:

1)如果直接使用安装的torchvision,会报以下错误:

RuntimeError: Couldn't load custom C++ ops. This can hap......

2)编译安装torchvision

(1)采用默认的visual studio 2019编译生成,好像对vs版本没啥要求!

(2)也会调用本机环境中的c++版的cuda10.2

python setup.py build develop
————————

pytorch-1.8.2+cuda10.2+torchvison-0.9.2(有问题)

pytorch/vision链接

1、拉vision代码

1)代码版本号需要跟pytorch对应上,也可以跟torchvison版本一致

2、重新编译生成torchvision:

1)如果直接使用安装的torchvision,会报以下错误:

RuntimeError: Couldn't load custom C++ ops. This can hap......

2)编译安装torchvision

(1)采用默认的visual studio 2019编译生成,好像对vs版本没啥要求!

(2)也会调用本机环境中的c++版的cuda10.2

python setup.py build develop