ubuntu 22.04 深度学习环境配置(pytorch版本)

发布时间 2023-08-10 15:00:51作者: zz子木zz

part1 验证方法

1 验证Anaconda

conda -V

2 验证显卡驱动

nvidia-smi

3 验证cuda

nvcc -V

4 验证pytorch、torchvision

conda list | grep torch

part2 安装顺序

1 装Anaconda、搞虚拟环境

conda create --name pku
conda create --name d2l-zh python=3.10

激活虚拟环境

conda activate chatglm_env

退出虚拟环境

conda deactivate

显示虚拟环境列表

conda info --env

2 装显卡驱动

上网搜教程,记得先关闭noveau驱动再安装Nvidia驱动

3 装cuda

上官网:https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=22.04&target_type=deb_network
image
image

4 装cudnn

还没搞,貌似是上官网(这里需要账号)安装文档安装即可
https://developer.nvidia.com/cudnn

5 装pytorch、torchvision(这个第一步做也可)

直接在虚拟环境中输入(我用默认源直接就成功了)

conda install pytorch
conda install torchvision