服务器miniconda安装pytorch

服务器miniconda安装使用gpu的pytorch

已成功安装并正常使用

使用nvidia-smi查看实验室cuda版本为11.4,但从/usr/local/cuda中查看到版本为11.2
我用到的安装命令为pytorch官网的安装命令,采用pip安装。
服务器miniconda安装pytorch_第1张图片
安装命令为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

你可能感兴趣的:(pytorch,服务器,深度学习)