pip报错,首先检查源的问题去用户目录下conda文件删除一些内容
官方推荐方式,但是这种方式由于墙的存在容易挂掉

cuda

cuda下载

版本依赖关系

注意:>=516.01指的是你的显卡驱动版本517.0、518.0都可以安装CUDA11.7GA版本的CUDA
7fe0263c-a4b8-4a3e-a495-5719fd426c05.png
image-20211018161652053
d41149ff-6773-4ac1-80bb-a2d4bc40db0c.png
104d57be-5c8f-49a3-9868-71fa302c6f7e.png

image-20211018162507982

torch

whl下载链接,下载完成直接pip install ***.whl
注意:aarch64表示arm64位适用于树莓派和Xavier的ARM架构的CPU

测试输出为true表示配置成功

import torch
print(torch.cuda.is_available())