每次换个环境又要安装pytorch,最开始安装总是出错,写下不踩坑安装全过程
1、首先看一下cuda的版本
运行命令 nvidia-smi
如下所示
比方说我的是右上角显示11.2,所以
2、去PyTorch 官网
选择Linux ,这里没有我的cuda版本
找过往版本Previous PyTorch Versions | PyTorch
发现有11.3 11.1版本和10.2版本,但是没有11.2版本,查阅得知可以向下兼容安装11.1版本的
3、往下滑找到cuda对应版本(11.1)版本的torch
按照以下命令安装
pip install torch==1.10.1+cu111 torchvision==0.11.2+cu111 torchaudio==0.10.1 -f https://download.pytorch.org/whl/cu111/torch_stable.html
等待……
安装成功
4、检查是否安装成功
以及确定是不是gpu版本的
输入python回车
文章来源:https://www.toymoban.com/news/detail-766325.html
返回true代表安装gpu版本的torch成功文章来源地址https://www.toymoban.com/news/detail-766325.html
到了这里,关于linux服务器安装gpu版本的pytorch的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!