安装cuda
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c nvidia
验证cuda与pytorch是否可用,输出为true可用
import torch
print(torch.cuda.is_available())
安装cuda
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c nvidia
验证cuda与pytorch是否可用,输出为true可用
import torch
print(torch.cuda.is_available())