栏目分类:
子分类:
返回
文库吧用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
文库吧 > IT > 软件开发 > 后端开发 > Python

CUDA安装教程

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

CUDA安装教程

 通过远程桌面连接服务器电脑。

 服务器自带驱动程序,无需自行下载。只需要下载CUDA、cuDNN和pytorch的3个文件即可。

  • 1 NVIDIA驱动版本和对应的CUDA版本查询

        驱动版本:451.77,对应CUDA版本为11.0.197(官网版本向下取) 

  • 2 CUDA下载

        CUDA下载官网:CUDA Toolkit Archive | NVIDIA Developer

          此处驱动版本可能不完全一致,但接近即可。

  • 3 cuDNN下载

        CuDNN官方下载网址:CUDA Deep Neural Network (cuDNN) | NVIDIA Developer

  • 4 安装CUDA和cuDNN

        默认安装地址,安装简易版即可。

        CUDA安装完成后,将cuDNN压缩包解压,改名为 cuDNN,复制到CUDA指定文件夹下:

  • 5 添加系统环境变量

        将CUDA文件夹下的

        C:Program FilesNVIDIA GPU Computing ToolkitCUDAv11.0extrasCUPTIlib64 和

        C:Program FilesNVIDIA GPU Computing ToolkitCUDAv11.0cudnnbin

        两个路径添加至系统变量path中

  • 6 CUDA&cuDNN安装成功检测

    均显示PASS即安装成功。

    三个版本查看(驱动设备、CUDA、cuDNN):

      

  • 7 Pytorch安装

        准备一个新建的工程文件,带有之前未安装过torch、torchvision等相关库的解释器。

        查询pytorch对应版本:Start Locally | PyTorch

         没有找到对应的CUDA 11.0版本,因此在previous版本中查看CUDA 11.0对应的torch版本:        找到对应的torch版本为1.7.1;torchvision版本为0.8.2;torchaudio版本为0.7.2

        Pytorch官方下载网址:https://download.pytorch.org/whl/torch_stable.html

        Ctr+F快速查询这三个待下载文件的对应版本进行下载:

  • 8 安装torch、torchvision和torchaudio

        将这三个文件复制粘贴至准备好的解释器scripts文件夹下:

        该文件夹路径下cmd进行三个库的安装:Pip install 文件名。

        例如:pip install torch-1.7.1+cu110-cp39-cp39-win_amd64.whl

        先安装torch,其余两个与torch安装方法相同。

  • 9 测试CUDA

     

        device=cuda,安装成功!

转载请注明:文章转载自 www.wk8.com.cn
本文地址:https://www.wk8.com.cn/it/1037115.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 wk8.com.cn

ICP备案号:晋ICP备2021003244-6号