pip安装pytorch的GPU版本

您所在的位置:网站首页 cuda版本下载 pip安装pytorch的GPU版本

pip安装pytorch的GPU版本

2023-03-01 22:41| 来源: 网络整理| 查看: 265

1.清华大学的anaconda开源镜像,根据anaconda搭配的python、pytorch、CUDA、cuDNN版本,然后依次寻找,单独下载进行安装;

2.安装NVIDIA图形驱动程序;

3.根据NVIDIA图形驱动的控制面板中,"系统信息"->"组件"->"NVCUDA.DLL"的信息内容下载CUDA版本;

安装时,需要确定当前电脑是否安装了Visual Studio软件,如果未安装则需要在安装阶段勾选Visual Studio的选项。

4.根据CUDA版本,下载对应的cuDNN(需要注意的是,下载时需要注册账户进行登录);

5.将下载的cuDNN压缩包进行解压,然后将压缩包中的内容依次复制到CUDA安装目录中;

(1) 复制 \cuda\bin\cudnn*.dll 到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\bin下 (2) 复制 \cuda\include\cudnn*.h 到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\include下 (3) 复制 \cuda\lib\x64\cudnn*.lib 到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\lib\x64下

6.在pytorch安装界面,选择对应的pytorch进行pip下载安装,如果下载速度较慢,则更改pip的镜像源;

7.安装完成后检查是否安装成功。

import torch torch.cuda.is_available() #显示True,则安装成功



【本文地址】


今日新闻


推荐新闻


CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3