pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
运行效果如下:
这个方法是直接从pytorch官网进行在线下载和安装。
cu121,表示当前您安装的cuda版本是12.1
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
运行效果如下:
这个方法是直接从pytorch官网进行在线下载和安装。
cu121,表示当前您安装的cuda版本是12.1