import os
os.system("nvidia-smi") #查看cuda版本号
#os.system("pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121")
os.system("python")
os.system("pip list")
torch.cuda.is_available() #检查是否安装好
纯代码安装pytorch-gpu版
最新推荐文章于 2024-05-22 17:13:59 发布