pytorch远程服务器安装
时间: 2023-11-07 22:03:30 浏览: 200
在Linux远程服务器上设置PyTorch的GPU版本
根据引用内容,如果你的服务器的CUDA最高版本为11.4,你可以选择安装PyTorch 1.12.1版本,并使用以下指令进行安装:conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch。如果你的服务器支持的CUDA最高版本高于11.6,你可以使用以下指令进行安装:conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia。如果你需要安装以前的版本,可以点击"install previous versions of PyTorch"。希望这些信息对你有帮助。
阅读全文