GIS之深度学习08：安装GPU环境下的pytorch

环境：

cuda：12.1.1

cudnn：12.x

pytorch：2.2.0

torchvision：0.17.0

Python：3.8

操作系统：win

---

（本文安装一半才发现pytorch与cuda未对应，重新安装了cuda后才开始的，请跳过！！）

首先创建虚拟环境：需要在anaconda prompt里运行

我这里创建的是py38版本

conda create -n gpupytorch python=3.8

大家可以根据自己的需要去下载，输入命令后点击回车，然后输入y，回车：

接着查看cuda的版本，cuda是允许多个版本并存的。其他版本可在官网上下载：

（我电脑自带的是cuad12.4,但是没找到对应的pytorch版本，于是重新安装了12.1.1版本）

cuda官网：CUDA 12.3 Update 2 发行说明 (nvidia.com)

查看pytorch与cuda对应的版本：Previous PyTorch Versions | PyTorch

由于版本未对应，这里重新安装Pytorch，小伙伴们也可以从这开始看。

重新激活环境

activate gpupytorch

传统的下载方法比较慢,我么可以先下载lunzi

地址：download.pytorch.org/whl/torch_stable.html

我的cuda版本为12.1.1，pytorch版本应为2.2.0，Python版本为3.8

检索适应的lunzi,首先是torch

（（torch文件已更正，为win版本，图片示例错误））

然后是torchvision

找一个位置存放lunzi（torch文件已更正，为win版本，图片示例错误）：

回到终端

链接文件存放的路径，并安装lunzi：

（以自己的路径为准）

(gpupytorch) C:\Users\Administrator>d:(gpupytorch) D:\>cd D:\PlugWheel

pip install torch-2.2.0+cpu-cp38-cp38-win_amd64.whl

pip install torchvision-0.17.0+cpu-cp38-cp38-win_amd64.whl

均显示“successfully”，则表示安装成功！！

检验：

打开anaconda，切换gpupytorch环境（cpu同理）