1. Anaconda 安装
(1)从官网下载文件,按照提示安装即可,建议不要安装到C盘。
官网地址:https://www.anaconda.com/download
重要的是配置下图所示的环境变量。依次点开 1 设置---> 2高级系统设置(对话框右侧)---> 2高级系统设置(对话框右侧)---> 3环境变量(对话框右下侧)---> 4 双击 Path ---> 5 新建添加三个环境变量。
(2)安装成功验证:
打开Anaconda3下面的Prompt,输入命令:conda --version,显示的是自己下载的版本则安装正确。
2.CUDA Pytorch
(1)查看本机GPU环境
此电脑(通过设置---个性化---主题----(右侧)桌面图标设置---选择 计算机)右键打开管理,查看驱动器版本(蓝线部分为驱动版本,目前的为556.12版本),这个版本决定了后续可安装的cuda等的版本。
如果没有相关的驱动信息,可以按照显卡型号,从NVIDIA GeForce 驱动程序的官网下载,网址为: https://developer.nvidia.com/cuda-toolkit-archive
(2)准备工作(非常重要)
查看CUDA和驱动版本的对应关系:
从官网 https://docs.nvidia.com/deploy/cuda-compatibility/index.html 上查看CUDA版本和驱动版本的对应关系。注意:一定要确保驱动和CUDA能够对应上。
pytorch与python的对应关系:从下面网址中查看
GitHub - pytorch/vision: Datasets, Transforms and Models specific to Computer Vision
pytorch 与cuda的版本对应关系:
建议依据cuda版本选择合适的pytorch版本,参考官网:https://pytorch.org/get-started/previous-versions/
(3)安装 pytorch cudnn cuda
在Ananconda Prompt使用如下命令进行安装(注意黄色部分的版本是需要依据自己机器进行选择的部分):
#conda create -n pytorch21 python=3.9 ##建立anaconda虚拟环境
#conda activate pytorch21
#安装pytorch、cuda、cudnn,命令来源于上图网址,选择适用于自己机器的版本号
#conda install pytorch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 pytorch-cuda=11.8 -c pytorch -c nvidia
(4)验证安装结果
在Ananconda Prompt使用如下命令进行验证
- conda activate pytorch21
- python
- import torch
- print(torch.cuda.is_available()) #返回True则说明已经安装了cuda
- #判断是否安装了cuDNN
- from torch.backends import cudnn
- print(cudnn.is_available()) #返回True则说明已经安装了cuDNN
- #判断cuda版本
- print(torch.__version__)
- print(torch.version.cuda)