因为是最新版所以官网还没有出对应的版本,故尝试上一版

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!

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐