yolov5_obb链接
1、使用netron查看yolov5s-best.onnx模型结构。
在终端运行(切换到模型文件所在路径)
pip install netron
python
import netron
netron.start(‘best.onnx’)#绝对地址用双反斜杠:\
如下图所示:
2、想要查看 Pytorch 实际使用的运行时的 cuda 目录,可以直接输出之前介绍的 cpp_extension.py 中的 CUDA_HOME 变量。
>>> import torch
>>> import torch.utils
>>> import torch.utils.cpp_extension
>>> torch.utils.cpp_extension.CUDA_HOME #输出 Pytorch 运行时使用的 cuda
另外查看环境中的cuda版本,打开cmd,打开环境,输入nvidia-smi
,或者nvcc --version
或者nvcc -V
3、关于切换cuda版本
计算机->右键,属性->高级系统设置->环境变量
如图,我想用11.1版本,把cuda_path改为11.1
然后点击path,将path中的关于11.1的向上移到所有cuda 版本的最前面,如下图:
一共两个关于11.1的,然后再看运行cuda版本就是11.1了,亲测有效。
4、windows下使用python setup.py install包含pytorch组件时报:文章来源:https://www.toymoban.com/news/detail-453896.html
cpp_extension.py:237: UserWarning: Error checking compiler version for cl: 'utf-8' codec can't decode byte 0xd3 in position 0: invalid continuation byte
解决方法(参考的)
打开cpp_extension.py文件,在第260行左右,有一段:文章来源地址https://www.toymoban.com/news/detail-453896.html
try:
if sys.platform.startswith('linux'):
minimum_required_version = MINIMUM_GCC_VERSION
version = subprocess.check_output([compiler, '-dumpfullversion', '-dumpversion'])
version = version.decode().strip().split('.')
else:
minimum_required_version = MINIMUM_MSVC_VERSION
compiler_info
到了这里,关于windows10安装yolov5_obb时的一些报错记录的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!