1. cuda
更新显卡驱动(windows下安装即可,和wsl共享):官方驱动 | NVIDIA
显卡驱动版本和cuda版本对应:CUDA 12.4 Update 1 Release Notes (nvidia.com)
下载cuda toolkit(注意Distribution需要选择wsl):CUDA Toolkit 12.4 Update 1 Downloads | NVIDIA Developer
2. cuDNN
cuDNN Archive | NVIDIA Developer
根据cuda版本,选择对应的cuDNN进行解压,解压缩后替换cuda目录同名文件夹
3. torch
pytorch/vision: Datasets, Transforms and Models specific to Computer Vision (github.com)
Previous PyTorch Versions | PyTorch
根据python版本,选择torch版本,再根据cuda版本选择合适的pytorch-cuda
4. vllm
因为vllm引入了其他cuda和torch版本,所以如果想使用vllm框架,则建议新建一个conda环境,再安装vllm
tips:清华镜像源
pip install xxxx -i Simple Indexhttps://pypi.tuna.tsinghua.edu.cn/simple