欢迎光临散文网 会员登陆 & 注册

Pytorch开发环境搭建

2023-02-17 22:09 作者:远-方-的-风  | 我要投稿

以windows作为开发环境

本人的笔记本配置情况:

CPU:AMD R7-6800H

显卡:RTX 3060 

系统:Windows 11

安装好Windows和必要的驱动后,需要安装的工具有:CUDA、Anaconda3、cuDNN、Pytorch-gpu

1. CUDA

CUDA是NVIDIA推出的运算平台。我们需要根据自己的显卡型号选择支持的CUDA版本,例如RTX 3060支持CUDA 12,下载地址:https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=11&target_type=exe_local

傻瓜式安装。

安装好后,需要再配置下系统的环境变量,电脑->鼠标右键->属性->高级系统设置->环境变量->Path:

这里我安的比较早,所以是CUDA11。

有时候傻瓜式安装完了就行~~~~~~

配置好后,就可以在cmd中使用nvidia-smi指令查看显卡了。

2. Anaconda3

Anaconda是Python的包管理器和环境管理器,可以方便我们安装Python的第三方库。


选择Windows下的Python 3.9的版本,安装也很简单,傻瓜式下一步即可。

安装好后,需要添加系统环境变量,方法与安装CUDA时一样。

路径改为自己安装的Anaconda路径即可。

配置好后,在cmd中运行conda -V没有报错,有版本信息输出,说明配置成功。

3. cuDNN和Pytorch安装

安装好Anaconda之后,使用conda安装cuDNN和Pytorch。

打开Anaconda Prompt,这是Anaconda自带的命令行工具,先一定要用这个工具创建环境,直接用系统自带的cmd,可能遇到一些奇怪的问题,例如CondaHTTPError的错误。在Anaconda Prompt中输入:

conda create -n your_name jupyter notebook

这句话的意思是创建一个名字为your_name的虚拟环境,并且这个虚拟环境额外安装jupyter notebook第三方库。可以将your_name改为你自己喜欢的名字,这个名字是你的虚拟环境的名字,自己随便取,比如LSP。

随后,输入y进行安装:

安装好后,可以通过指令conda info -e查看已有环境情况:

从上图可以看到,有四个环境,一个是base,自带的基础环境,其他三个是创建的新环境。新建环境可以分开管理我们配置的环境。

安装好环境后,我们就可以激活xzhy环境,并安装cuDNN和GPU版的Pytorch了。激活名为xzhy的环境:activate xzhy

可以看到,我们的环境由base变成了xzhy。在xzhy环境中安装cuDNN:conda install cudnn

安装cuDNN好后,安装Pytorch,打开Pytorch官网:https://pytorch.org/


根据自己的环境选择,选择好后,网页会自动给出需要运行的指令。这里可能需要区分下Python的版本和CUDA的版本。

Python版本查看方法:直接在命令行中输入python,会看到Python的版本。

CUDA版本查看方法,在命令行中输入nvidia-smi:

确定好版本后,就可以通过Pytorch官网提供的指令安装GPU版本的Pytorch了。

至此,基础的环境搭建已经完成。






Pytorch开发环境搭建的评论 (共 条)

分享到微博请遵守国家法律