pip安装pytorch的GPU版本 |
您所在的位置:网站首页 › cuda版本下载 › pip安装pytorch的GPU版本 |
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 |