环境安装配置
1. 安装虚拟环境Conda
下载安装anaconda:
初始化conda环境
conda create -n <环境名称> python=<版本号>(python版本>=3.8)
激活当前环境
-
conda activate <环境名称> -
查看Pytorch版本
pip list
2. 安装CUDA
命令行输入 nvidia-smi 查看CUDA支持的最高版本 (仅支持Nvidia显卡)
为了确保版本兼容,请先删除nvidia所有的文件
下载安装版本CUDA
下载安装cuDNN,选择对应的版本:(如:我是CUDA 11.6,选择CUDA 11.X的均可,建议不要选最新版本以免引起兼容问题)
拷贝这三个文件
粘贴到此目录
3. 安装PyTorch (命令需要包含+cu字样)
Previous PyTorch Versions | PyTorch
查看pytorch是否安装成功
pip list
下一章节讲解如何使用ultralytics(Yolov8)