root@ai-box-Wan2:/workspace# python -c "import torch; print(torch.version); print(torch.ones(2).cuda())"
2.4.0+metax3.3.0.2
tensor([1., 1.], device='cuda:0')
root@ai-box-Wan2:/workspace# python3 11.py
PyTorch版本: 2.4.0+metax3.3.0.2
是否包含nms算子: False
root@ai-box-Wan2:/workspace# cat 11.py
import torch
print(f"PyTorch版本: {torch.version}")
print(f"是否包含nms算子: {'torchvision::nms' in dir(torch.ops)}")
roroot@ai-box-Wan2:/workspace# echo MACA_PATH
/opt/maca
root@ai-box-Wan2:/workspace# echo LD_LIBRARY_PATH
/opt/maca/lib:/opt/maca/ompi/lib:/opt/maca/ucx/lib:/opt/mxdriver/lib:
root@ai-box-Wan2:/workspace# python3 33.py
Traceback (most recent call last):
File "/workspace/33.py", line 4, in <module>
import torchvision.transforms as T
ModuleNotFoundError: No module named 'torchvision'
root@ai-box-Wan2:/workspace# cat 33.py
import torch
import numpy as np
from PIL import Image
import torchvision.transforms as T
img = Image.fromarray(np.random.randint(0, 255, (100, 100, 3), dtype=np.uint8))
transform = T.Compose([
T.Resize(256),
T.CenterCrop(224),
T.ToTensor(),
])
tensor_img = transform(img)
print("图像处理成功，张量形状:", tensor_img.shape)
root@ai-box-Wan2:/workspace# python3 test.py
Traceback (most recent call last):
File "/workspace/test.py", line 2, in <module>
import torchvision.ops as ops
ModuleNotFoundError: No module named 'torchvision'
root@ai-box-Wan2:/workspace# cat test.py
import torch
import torchvision.ops as ops
boxes = torch.tensor([[10, 10, 50, 50], [20, 20, 60, 60]], dtype=torch.float)
scores = torch.tensor([0.9, 0.8])
keep = ops.nms(boxes, scores, 0.5)
print("NMS运行成功，保留的框索引:", keep)