ubuntu安装pytorch-gpu

安装miniconda

  1. 到miniconda的官网下载最新版本的miniconda
    ubuntu安装pytorch-gpu_第1张图片
  2. 将下载好的miniconda安装包上传至ubuntu服务器上
  3. 安装miniconda,敲yes,默认安装即可
sh Miniconda3-latest-Linux-x86_64.sh
  1. 使用conda命令创建虚拟环境torch
conda create -n torch python=3.9
  1. 激活刚刚创建的torch环境
conda activate torch

安装nvidia显卡驱动

查看驱动是否安装

nvidia-smi

出现以下报错说明没有显卡驱动
ubuntu安装pytorch-gpu_第2张图片

查看显卡可用的驱动

ubuntu-drivers devices

ubuntu安装pytorch-gpu_第3张图片

安装显卡驱动

#自动安装显卡驱动
sudo ubuntu-drivers autoinstall

查看显卡驱动

nvidia-smi

显卡版本是515. CUDA支持11.7及以下
ubuntu安装pytorch-gpu_第4张图片

安装pytorch

记得先激活刚刚创建的torch环境

conda activate torch

然后打开pytorch的官网,根据自己的情况选择安装命令
ubuntu安装pytorch-gpu_第5张图片

复制命令,安装pytorch

中途可能会断开,可以多是几次。如果网速慢,可以尝试配置国内的conda源

conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge

验证gpu是否可用

python
import torch
torch.cuda.is_available()

返回True,说明gpu可用
在这里插入图片描述
至此,pytorch-gpu环境搭建完成
后面可以安装jupyter、和numpy、pandas,在jupyternotebook中进行开发吧

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