在给分割工具配环境的时候
发现Apex编译失败,发现是Cuda版本的问题,于是按照博客的思路更改setup文件中的校验部分做如下修改:
在setup.py文件中,找到get_cuda_bare_metal_version函数,在
bare_metal_minor = release[1][0]
后添加
bare_metal_minor = '0'
强行将系统校验的cuda版本变为11.0
本系统的具体环境如下:
pytorch = 1.7.1
python = 3.7.12
cudatoolkit = 11.0.3
nvcc --version为11.5
注:
1.在python中import torch检查pytorch是否正确安装。
2.查看安装的pytorch是否是gpu版本