草庐IT

GPU版本PyTorch详细安装教程

吉果果、 2023-05-03 原文

目录

一、安装显卡驱动

1、查看显卡驱动型号

2、下载显卡驱动

3、查看GPU状态

二、安装Visual Studio 2019

三、安装CUDA

1、下载对应版本的CUDA

2、安装下载好的CUDA

3、设置环境变量

 四、安装cudnn

五、安装anaconda

六、安装PyTorch

1、创建虚拟环境

2、激活并进入虚拟环境

3、安装PyTorch

4、验证PyTorch是否安装成功

注意:30系列的的显卡暂时不支持cuda11以下版本!!!

一、安装显卡驱动

1、查看显卡驱动型号

右击右下角开始,在设备管理器中查看计算机显卡型号,例如我的显卡是GTX1050:

2、下载显卡驱动

进入英伟达官网,下载对应的显卡驱动:官方驱动 | NVIDIAhttps://www.nvidia.cn/Download/index.aspx?lang=cn

选择对应版本参数并下载安装,安装时直接一直下一步即可。

3、查看GPU状态

安装完成后,打开CMD终端(快捷键:Win+R),命令行输入:nvidia-smi查看GPU状态:

 注意:右上角显示的CUDA版本(红色框中)是当前版本所支持的最高CUDA版本。

二、安装Visual Studio 2019

由于要使用cuda则必须要有一个编译工具,这里安装的是Visual Studio 2019

下载Visual Studio 社区版链接https://visualstudio.microsoft.com/zh-hans/downloads/

安装完成后进入以下界面,注意:勾选“Python开发“C++桌面开发

三、安装CUDA

1、下载对应版本的CUDA

 CUDA各版本官方下载地址: CUDA Toolkit Archive | NVIDIA Developerhttps://developer.nvidia.com/cuda-toolkit-archive

 CUDA用的是10.2版本,因此进入cuda官网找到对应的版本进行下载:

 

2、安装下载好的CUDA

下载完成后将CUDA安装,注意:建议默认路径,后期需要添加环境变量,安装时选择自定义安装

勾选Visual Studio Integration:

3、设置环境变量

鼠标右键计算机(此电脑),打开属性->高级系统设置->环境变量,可以看到系统变量中多了CUDA_PATHCUDA_PATH_V10_2两个环境变量。

接下来,还要在系统变量中“新建”,添加以下几个环境变量,以下是默认安装位置的路径的环境变量(左边是变量名,右边是变量值):

CUDA_SDK_PATH = C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.2

CUDA_LIB_PATH = %CUDA_PATH%\lib\x64

CUDA_BIN_PATH = %CUDA_PATH%\bin

CUDA_SDK_BIN_PATH = %CUDA_SDK_PATH%\bin\win64

CUDA_SDK_LIB_PATH = %CUDA_SDK_PATH%\common\lib\x64

在系统变量 Path 的末尾添加:

%CUDA_LIB_PATH%;%CUDA_BIN_PATH%;%CUDA_SDK_LIB_PATH%;%CUDA_SDK_BIN_PATH%;

双击Path,再添加如下5(默认安装路径):C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\lib\x64

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\include

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\extras\CUPTI\lib64

C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.2\bin\win64

C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.2\common\lib\x64

 

四、安装cudnn

1、下载cuda10.2对应的cudnn

       cudnn下载地址https://developer.nvidia.com/cudnn

2、下载cudnn后直接将其解开压缩包,然后需要将解压后的bin,include,lib文件夹复制粘贴到cuda安装时的默认路径文件夹下(比如我的是在:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2 下)

注意:对整个文件夹bin,include,lib复制粘贴

3、最后测试cuda是否配置成功:

Win+R打开CMD终端执行:nvcc -V 即可看到cuda的信息:

五、安装anaconda

详细安装教程可参考以下链接:

(42条消息) Python安装教程步骤1:Pycharm和Anaconda的安装_ZHW_AI课题组的博客-CSDN博客https://blog.csdn.net/m0_37758063/article/details/116540837?spm=1001.2014.3001.5502

六、安装PyTorch

1、创建虚拟环境

在anaconda中添加镜像源:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge 
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

在开始栏打开刚才安装的anaconda prompt 创建项目运行虚拟环境:

conda create -n your_env_name(虚拟环境名称) python==xx(想要创建的虚拟环境的python版本号)

注:若不知道python版本,Win+R打开CMD终端执行:python,然后enter即可看到所安装的python版本号。

可以根据自己需要修改,例如:conda create -n mytorch python==3.9.7

输入y,然后按下enter确认开始下载安装。

出现以上界面则虚拟环境已经创建完成。

2、激活并进入虚拟环境

1、在上一步的基础上,对创建完成的虚拟环境进行激活,在anaconda prompt 终端中输入以下指令:

conda activate mytorch

2、若要退出当前的虚拟环境,执行以下指令即可:

conda deactivate

3、安装PyTorch

1、此时我们已经激活并进入到所创建的mytorch环境中。 

2、接下来进入PyTorch官网,选择相关参数,获取PyTorch安装指令,并在anaconda prompt 终端中执行红色框中的指令(该指令为最新版本的PyTorch)

注意:安装的时候要将命令后的-c pytorch后面的内容删除,从国内源进行下载,速度快一些。PyTorchhttps://pytorch.org/

另:若需要选择其他版本的PyTorch点击绿色框中的指令即可 

然后进入以下页面,找到需要的版本进行安装即可

3、输入PyTorch安装指令指令后,查看conda找到将要安装的包是否是要进行安装的

如果是,输入y之后,按下enter确认进行下载;

如果不是,就需要重新检查安装PyTorch的命令。

3、 安装PyTorch,至此,基础环境已经部署完成。

4、验证PyTorch是否安装成功

1、安装完成后,继续在在anaconda prompt 终端中执行以下指令,验证PyTorch是否安装成功

python 
import torch 
torch.cuda.is_available() 

2、如下图提示True说明框架配置成功,且GPU可用

 3、验证完成后Ctrl+Z回到命令行,然后执行conda list指令就可以看到该虚拟环境下已经安装好的包

能够在conda list中找到以上两个包,则说明已经配置完成。

有关GPU版本PyTorch详细安装教程的更多相关文章

  1. ruby - 在 64 位 Snow Leopard 上使用 rvm、postgres 9.0、ruby 1.9.2-p136 安装 pg gem 时出现问题 - 2

    我想为Heroku构建一个Rails3应用程序。他们使用Postgres作为他们的数据库,所以我通过MacPorts安装了postgres9.0。现在我需要一个postgresgem并且共识是出于性能原因你想要pggem。但是我对我得到的错误感到非常困惑当我尝试在rvm下通过geminstall安装pg时。我已经非常明确地指定了所有postgres目录的位置可以找到但仍然无法完成安装:$envARCHFLAGS='-archx86_64'geminstallpg--\--with-pg-config=/opt/local/var/db/postgresql90/defaultdb/po

  2. ruby - 完全离线安装RVM - 2

    我打算为ruby​​脚本创建一个安装程序,但我希望能够确保机器安装了RVM。有没有一种方法可以完全离线安装RVM并且不引人注目(通过不引人注目,就像创建一个可以做所有事情的脚本而不是要求用户向他们的bash_profile或bashrc添加一些东西)我不是要脚本本身,只是一个关于如何走这条路的快速指针(如果可能的话)。我们还研究了这个很有帮助的问题:RVM-isthereawayforsimpleofflineinstall?但有点误导,因为答案只向我们展示了如何离线在RVM中安装ruby。我们需要能够离线安装RVM本身,并查看脚本https://raw.github.com/wayn

  3. ruby-on-rails - rails 目前在重启后没有安装 - 2

    我有一个奇怪的问题:我在rvm上安装了ruby​​onrails。一切正常,我可以创建项目。但是在我输入“railsnew”时重新启动后,我有“程序'rails'当前未安装。”。SystemUbuntu12.04ruby-v"1.9.3p194"gemlistactionmailer(3.2.5)actionpack(3.2.5)activemodel(3.2.5)activerecord(3.2.5)activeresource(3.2.5)activesupport(3.2.5)arel(3.0.2)builder(3.0.0)bundler(1.1.4)coffee-rails(

  4. ruby - 如何为 emacs 安装 ruby​​-mode - 2

    我刚刚为fedora安装了emacs。我想用emacs编写ruby。为ruby​​提供代码提示、代码完成类型功能所需的工具、扩展是什么? 最佳答案 ruby-mode已经包含在Emacs23之后的版本中。不过,它也可以通过ELPA获得。您可能感兴趣的其他一些事情是集成RVM、feature-mode(Cucumber)、rspec-mode、ruby-electric、inf-ruby、rinari(用于Rails)等。这是我当前用于Ruby开发的Emacs配置:https://github.com/citizen428/emacs

  5. ruby-on-rails - 无法在centos上安装therubyracer(V8和GCC出错) - 2

    我正在尝试在我的centos服务器上安装therubyracer,但遇到了麻烦。$geminstalltherubyracerBuildingnativeextensions.Thiscouldtakeawhile...ERROR:Errorinstallingtherubyracer:ERROR:Failedtobuildgemnativeextension./usr/local/rvm/rubies/ruby-1.9.3-p125/bin/rubyextconf.rbcheckingformain()in-lpthread...yescheckingforv8.h...no***e

  6. ruby - 通过 RVM (OSX Mountain Lion) 安装 Ruby 2.0.0-p247 时遇到问题 - 2

    我的最终目标是安装当前版本的RubyonRails。我在OSXMountainLion上运行。到目前为止,这是我的过程:已安装的RVM$\curl-Lhttps://get.rvm.io|bash-sstable检查已知(我假设已批准)安装$rvmlistknown我看到当前的稳定版本可用[ruby-]2.0.0[-p247]输入命令安装$rvminstall2.0.0-p247注意:我也试过这些安装命令$rvminstallruby-2.0.0-p247$rvminstallruby=2.0.0-p247我很快就无处可去了。结果:$rvminstall2.0.0-p247Search

  7. ruby-on-rails - 项目升级后 Pow 不会更改 ruby​​ 版本 - 2

    我在我的Rails项目中使用Pow和powifygem。现在我尝试升级我的ruby​​版本(从1.9.3到2.0.0,我使用RVM)当我切换ruby​​版本、安装所有gem依赖项时,我通过运行railss并访问localhost:3000确保该应用程序正常运行以前,我通过使用pow访问http://my_app.dev来浏览我的应用程序。升级后,由于错误Bundler::RubyVersionMismatch:YourRubyversionis1.9.3,butyourGemfilespecified2.0.0,此url不起作用我尝试过的:重新创建pow应用程序重启pow服务器更新战俘

  8. ruby - 如何在 Lion 上安装 Xcode 4.6,需要用 RVM 升级 ruby - 2

    我实际上是在尝试使用RVM在我的OSX10.7.5上更新ruby,并在输入以下命令后:rvminstallruby我得到了以下回复:Searchingforbinaryrubies,thismighttakesometime.Checkingrequirementsforosx.Installingrequirementsforosx.Updatingsystem.......Errorrunning'requirements_osx_brew_update_systemruby-2.0.0-p247',pleaseread/Users/username/.rvm/log/138121

  9. ruby - Fast-stemmer 安装问题 - 2

    由于fast-stemmer的问题,我很难安装我想要的任何ruby​​gem。我把我得到的错误放在下面。Buildingnativeextensions.Thiscouldtakeawhile...ERROR:Errorinstallingfast-stemmer:ERROR:Failedtobuildgemnativeextension./System/Library/Frameworks/Ruby.framework/Versions/2.0/usr/bin/rubyextconf.rbcreatingMakefilemake"DESTDIR="cleanmake"DESTDIR=

  10. ruby-on-rails - 在 ruby​​ .gemspec 文件中,如何指定依赖项的多个版本? - 2

    我正在尝试修改当前依赖于定义为activeresource的gem:s.add_dependency"activeresource","~>3.0"为了让gem与Rails4一起工作,我需要扩展依赖关系以与activeresource的版本3或4一起工作。我不想简单地添加以下内容,因为它可能会在以后引起问题:s.add_dependency"activeresource",">=3.0"有没有办法指定可接受版本的列表?~>3.0还是~>4.0? 最佳答案 根据thedocumentation,如果你想要3到4之间的所有版本,你可以这

随机推荐