Linux安装detectron2出现nvcc fatal : Unsupported gpu architecture compute_75
我的电脑python3.6.9,pytorch1.4,opencv4.2,具体安装方式可以自行百度
参考链接
基本配置及版本要求
1.OS: Linux 或 macOX
2.Python>=3.6
3.PyTorch>=1.4
4.opencv(python 版)(可选),用于demo运行和可视化
5.安装pycocotools
查看pytorch是否能调用GPU
import torch, torchvision
print(torch.__version__, torch.cuda.is_a
原创
2020-05-31 16:37:42 ·
1331 阅读 ·
0 评论