赞
踩
因为是最新版所以官网还没有出对应的版本,故尝试上一版
conda install pytorch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 pytorch-cuda=12.1 -c pytorch -c nvidia
验证torch能否正常使用以及版本
torch进行简单的计算
验证能否使用gpu
OK!