1. 安装anaconda
到官网下载最新版的anaconda,下载对应的windows版本,地址:anaconda官网
下载后直接安装,安装完成后配置环境变量,具体可以百度anaconda安装说明。安装完成后,打开cmd,输入conda --version,如果能查看到版本信息,说明安装成功。
2. 给conda配置镜像源。
3. 安装cuda
在官网下载cuda安装,地址:CUDA Toolkit 12.2 Downloads | NVIDIA Developer
我下载的12.2版本,安装后通过cmd输入nvcc -V可查看是否安装成功
nvidia-smi
3. 创建环境
1)conda create -n pygpu python=3.8 创建名为pygpu的环境,安装python3.8版
2)conda activate pygpu 激活环境
3)根据官网命令安装pytorch,地址:pytorch官网
命令:
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
4)安装完成后,检查pytorch-gpu版是否安装成功
True表示 pytorch-gpu安装成功。