确保本地显卡是N卡,已经按照最新驱动
下载并安装: cuda_12.6.3_561.17_windows
下载并安装: cudnn-windows-x86_64-9.3.0.75_cuda12-archive
下载并安装: Miniconda3-latest-Windows-x86_64(最好选中设置环境变量,虽然它不建议),修改源为清华,具体忘了哪个命令。
conda create -n torch
conda activate torch
进入https://pytorch.org/get-started/previous-versions/,找到Linux and Windows,看到下面的命令
# CUDA 12.4
pip install torch==2.5.0 torchvision==0.20.0 torchaudio==2.5.0 --index-url https://download.pytorch.org/whl/cu124
所以参照上面,在conda中输入
conda install pytorch2.5.0 torchvision0.20.0 torchaudio==2.5.0 pytorch-cuda=12.4 -c pytorch -c nvidia
完成之后,python
输入下面代码,显示版本和是否支持CUDA
import torch
print(torch.version) #显示版本
print(torch.cuda.is_available()) #
另外的命令
conda env list