服务器miniconda安装pytorch

原创已于 2022-03-18 19:52:32 修改 · 1.3k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #服务器 #深度学习

于 2021-12-23 16:37:35 首次发布

本文介绍如何在服务器上安装能利用GPU资源的PyTorch，并确保其正常运行。文章详细记录了作者通过pip安装特定版本PyTorch的过程，包括torch、torchvision和torchaudio等组件。同时，验证了CUDA支持的有效性。

服务器miniconda安装使用gpu的pytorch

已成功安装并正常使用

使用nvidia-smi查看实验室cuda版本为11.4，但从/usr/local/cuda中查看到版本为11.2
我用到的安装命令为pytorch官网的安装命令，采用pip安装。
在这里插入图片描述
安装命令为pip3 install torch==1.10.1+cu113 torchvision==0.11.2+cu113 torchaudio==0.10.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
我科学上网，就直接可以下载安装，中间也没有出现问题。
最后torch.cuda.is_available()返回True