1 下载anaconda
官网:Index of /anaconda/archive/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror
选择下载Anaconda-1.4.0-Windows-x86.exe
2 创建虚拟环境
以管理员身份打开Anaconda Prompt
conda env list
conda creat -n yolov8 python=3.8
创建过程中有提示,填y即可
3 下载CUDA
前置条件:查看系统支持CUDA的版本
win+r,输入cmd回车,进入命令行输入nvidia-smi,再输入nvcc -V
cuda下载官网:CUDA 工具包档案 |NVIDIA 开发人员
4 安装pytorch
前置条件:配置镜像源
以管理员身份打开Anaconda Powershell Prompt
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
conda config --get
4.1官网获取下载命令
pytorch下载官网Start Locally | PyTorch
官网上复制命令
conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia
4.2下载pytorch
以管理员身份打开Anaconda Prompt
激活虚拟环境命令:
conda activate yolov8
下载pytorch命令:
conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia
输入conda list,看有没有pytorch或者torch
5 pycharm使用pytorch