赞
踩
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组件时报:
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行左右,有一段:
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
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。