本篇文章是在安装好anaconda的前提下进行的后续操作噢
第一:确定自己的显卡版本号和支持最大的cuda号:
①在电脑上打开自己的显卡驱动控制面板:
进入帮助,点击系统信息
依据自己的驱动版本(我的是419.72),找支持最大的cuda号:
我的最大支持CUDA 10.1,你可以找对应的
②查看自己的python版本号:
可以进入Anaconda Prompt (anaconda):
后输入:python -V
第二:确定好后就可以进行pytorch安装的准备工作
①我先装的是cuda,你可以查看一下你的电脑是否已经自己安装好了支持最大的版本的cuda,查看方式:在显卡驱动控制面板的系统信息里的组件中。
②如果不是则进行安装,是的话就跳过这个步骤:
1.直接在官网选择对应版本下载(也可以用驱动精灵更方便一点)
2.官网为:CUDA Toolkit Archive
3.察看自己对应的版本进行下载
4.下载
5.安装流程
下载好了直接双击安装就可以,路径直接默认就好,等待安装界面出现,“同意”。
选精简,然后一步步安装即可,等待安装完成就完了
第三:寻找pytorch
依据自己所安装的CUDA,去选择合适的pytorch版本:
搜索: PyTorchhttps://pytorch.org/然后寻找自己对应的安装包所对应的命令,因为是anaconda环境所以选择anaconda所对应的安装命令:
比如我的,找到的对应CUDA10.1的并且用anaconda安装的是这个pytorch=1.7.1的版本,之后开是进入安装步骤:
第四:安装pytorch:
①进入Anaconda Prompt (anaconda):
②进入自己的anaconda环境:
以我的为例:
③输入命令进入自己的anaconda里面,在进入enve
④创建新的环境命令:
conda create -n pytorch-gpu python=3.8
命令解释:pytorch-gpu 是你的新环境名字 python是你的python版本,按自己的改一下就行
创建完成
⑤进入所创建的环境:
输入:conda info --envs 查看环境是否已经好了
我的已经好了
在进入自己的环境里边,输入:activate pytorch-gpu
可以看到base已经变成了你的环境名字
⑥输入安装pytorch命令进行安装,这个命令在第三:寻找pytorch已经找到,我的对应的是:
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch
开始安,会慢一点:
遇到这个输入:y
安装完成:
第五:验证安装
输入 python
输入 import torch
输入 torch.__version__
大功告成,撒花