环境:
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同理)