ubuntu 快速安装 pytorch

一、pytorch环境搭建

从官网pip安装速度太慢,这里介绍一种快速安装方法,首先打开在终端中输入

nvidia-smi

查看cuda版本,注意11.4的可以装11.1 11.2 不一定得是11.4;

其次去pytorch官网确定要安装的pytorch ,例如安装 :

pip install torch==1.10.0+cu111 torchvision==0.11.0+cu111 torchaudio==0.10.0 -f https://download.pytorch.org/whl/torch_stable.html

接着去pytorch whl源 ,下载要安装的torch、torchvision、torchaudio,注意版本要对应;

下载完成直接 pip install xxx/.whl;

最后检测pytorch安装成功

import torch
torch.cuda.is_available()
返回True则表示安装成功

你可能感兴趣的:(人工智能,ubuntu,深度学习,pytorch)