pytorch-1.8.2+cuda10.2+torchvison-0.9.2(有问题)
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
标签:torchvision,模型,torchvison,编译,pytorch,win10,cuda10.2 From: https://www.cnblogs.com/wllwqdeai/p/16926399.html