linux anconda下基础环境配置(torch、opencv等)
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
原文地址:https://blog.csdn.net/weixin_45702256/article/details/142596456
免责声明:本站文章内容转载自网络资源,如本站内容侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!