YOLO安装

因为是第三方，下载离线包

wget https://mirrors.aliyun.com/anaconda/pkgs/main/win-64/cudatoolkit-11.3.1-h59b6b97_2.tar.bz2?spm=a2c6h.25603864.0.0.6b013eb1LvLRjn
wget https://mirrors.aliyun.com/anaconda/cloud/pytorch/win-64/pytorch-1.12.0-py3.8_cuda11.3_cudnn8_0.tar.bz2?spm=a2c6h.25603864.0.0.77042797kaedTY

conda安装

conda install --offline "F:\download\pytorch-1.12.0-py3.8_cuda11.3_cudnn8_0.tar.bz2"
conda install --offline "F:\download\cudatoolkit-11.3.1-h59b6b97_2.tar.bz2"

测试安装是否成功

import torch

if __name__ == '__main__':
    print(torch.cuda.is_available())
    print(torch.backends.cudnn.is_available())
    print(torch.cuda_version)
    print(torch.backends.cudnn.version())