1、torch安装(GPU)
下载链接:https://pytorch.org/
根据配置下载对应版本,CUDA11.4 可用11.3下的安装包
conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.3 -c pytorch
错误解决:
安装后import torch
时报错
解决方式:
sudo rmmod nvidia_uvm
sudo modprobe nvidia_uvm
2、opencv安装
# 打开虚拟环境
setconda
# 激活虚拟环境
conda activate yolov8
# 安装opencv
pip install opencv-python
pip install opencv-contrib-python