anaconda安装pytorch-gpu版本

1.安装cuda环境

1.1 更新显卡驱动(非必需,只要步骤1.2中的CUDA版本支持11.6及以上就可以)

打开NVIDIA控制面板,可以看到显卡型号:GeForce RTX 2060,显卡驱动版本:512.78

anaconda安装pytorch-gpu版本_第1张图片

进入NVIDIA官网检索最新驱动版本并更新(我偷懒没更新):https://www.nvidia.cn/Download/index.aspx?lang=cn

anaconda安装pytorch-gpu版本_第2张图片

1.2 查看显卡对应的CUDA版本

nvidia控制面板->帮助->系统信息->组件,可以看到CUDA版本是11.6.134

anaconda安装pytorch-gpu版本_第3张图片
anaconda安装pytorch-gpu版本_第4张图片

1.3 下载并安装对应版本的CUDA,推荐本地安装(local)

https://developer.nvidia.com/cuda-toolkit-archive

anaconda安装pytorch-gpu版本_第5张图片

1.4 下载并安装对应版本的cuDNN,需要注册一个账号

https://developer.nvidia.com/rdp/cudnn-archive

anaconda安装pytorch-gpu版本_第6张图片

2. 下载并安装anaconda

https://www.anaconda.com/

anaconda安装pytorch-gpu版本_第7张图片

2.1 在conda虚拟环境中安装python

打开Anaconda Powershell Prompt

anaconda安装pytorch-gpu版本_第8张图片

输入命令conda create -n env_name python=x.x

conda create -n pytorch python=3.7

……

3. 安装pytorch

方法一:进入官网找到对应版本的安装命令,如果网速很慢的话会需要换源

https://pytorch.org/get-started/locally/

anaconda安装pytorch-gpu版本_第9张图片

方法二:离线下载安装包到本地,再进行离线安装

https://download.pytorch.org/whl/torch_stable.html

anaconda安装pytorch-gpu版本_第10张图片

……

4. 安装pycharm专业版,使用学校邮箱或学信网在线报告进行认证免费使用,一年认证一次。配置pycharm项目的解释器。

你可能感兴趣的:(软件安装,pytorch,深度学习,python)