先说几个方法:
1)我的服务器上有一张卡不能用(亮机卡),于是在直接在要运行的那个文件前:
import os
os.environ['CUDA_VISIBLE_DEVICES']='0'
没用。
2)网上做法:
https://github.com/NVIDIA/apex/issues/615
于是我在anaconda3/envs/byteTrack/lib/python3.8/site-packages/torch/utils/cpp_extension.py中大概1577行中supported_arches添加了“3.0”。
结果:不报这个错,报别的错了。
3)大佬指点
别使用1)的做法,直接运行命令中指定
CUDA_VISIBLE_DEVICES=0 python setup.py develop
(第二次碰到了,遇到就忘了怎么处理,得亏大佬指点
(0是我可以使用的卡号,1是亮机卡号,使用nvidia-smi就可以查看所有卡号