甘肃建设职工教育培训中心网站,计算机网络设计是干什么的工作,导航网站建站系统,国内跨境电商平台有哪些?文章目录 前言一、安装cudn二、安装cudnn三、安装pytorch 前言 确保Windows系统版本高于windows10 21H2或Windows11#xff0c;然后在Windows中将显卡驱动升级到最新即可#xff0c;WSL2已支持对显卡的直接调用。 一、安装cudn 进入英伟达官网中的cuda下载地址#xff1… 文章目录 前言一、安装cudn二、安装cudnn三、安装pytorch 前言 确保Windows系统版本高于windows10 21H2或Windows11然后在Windows中将显卡驱动升级到最新即可WSL2已支持对显卡的直接调用。 一、安装cudn 进入英伟达官网中的cuda下载地址CUDA Toolkit 11.7 Update 1 Downloads | NVIDIA Developer 由于作者使用的是WSL-Ubuntu20.04长期支持版所以按照以下配置来进行安装。 作者采用的是runfilelocal下载方式因为此方式简单原理为通过wget命令下载cuda安装包通过sh命令解压安装cuda wget https://developer.download.nvidia.com/compute/cuda/11.7.1/local_installers/cuda_11.7.1_515.65.01_linux.run sudo sh cuda_11.7.1_515.65.01_linux.run程序会进行解压和安装速度会比较慢最后会出现如下界面输入accept回车确认 之后便会出现如下界面前面带有[X]的表示你要安装的工具最后选择Install回车确认 安装完成后通过修改bashrc文件配置来让cuda更好的运行 使用vim打开配置文件 sudo vi ~/.bashrc将如下代码加入到文件的最后几行 export PATH/usr/local/cuda/bin:$PATH export LD_LIBRARY_PATH/usr/local/cuda/lib64:$LD_LIBRARY_PATH使配置文件生效 source ~/.bashrc使用nvcc -V查看cuda是否安装成功成功会如下显示 二、安装cudnn 因为作者使用的是WSL2-Ubuntu20.04,所以进入cuDNN下载地址cuDNN Archive | NVIDIA Developer中选择如下安装包(注意cudnn的版本必须与安装的cuda百分百对应)并将此安装包复制到Ubuntu下的/home/文件夹下 通过如下命令解压cudnn包使用tar xvf命令 tar xvf cudnn-linux-x86_64-8.9.2.26_cuda11-archive.tar.xz然后通过cp复制命令将解压好的cuDNN文件复制到cuda配置文件中 即配置cuda/include下所有文件和cuda/lib64下所有文件 运行如下代码即可 sudo cp cudnn-linux-x86_64-8.9.2.26_cuda11-archive/include/* /usr/local/cuda/include/ sudo cp cudnn-linux-x86_64-8.9.2.26_cuda11-archive/lib/* /usr/local/cuda/lib64/至此cudnn已经安装完成。 三、安装pytorch 安装地址https://pytorch.org/get-started/previous-versions/找到1.13.1版本选择对应cuda版本安装 输入以下命令即可 pip install torch1.13.1cu117 torchvision0.14.1cu117 torchaudio0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117测试 import torch print(torch.__version__)#版本号 print(torch.cuda.is_available())#显卡是否可用 print( torch.cuda.get_device_name())#显卡型号