Python版本:3.7
1.安装任意版本的Anaconda
首先安装anaconda Windows下安装anaconda
进入文件夹,选择对应的anaconda版本,选择如下版本为python3.7(要安装其他的就选其他版本号) Anaconda其它版本文件夹
安装完成后,在搜索框里输入cmd,打开terminal看看有木有安装成功
输入 python -V 当当当当当当~ 3.7版本安装成功啦
2.配置显卡NVIDIA
打开任务管理器,ctr+Alt+Del,然后看看是否显示是它,是就说明驱动已经存在的,不是的话去安装一下驱动就行。 如果没有GPU的话,注意哈,没有的话直接省略此步鄹,直接进入步鄹4
3.查看cuda版本
打开NVIDIA控制面板,咋个打开呢,可以在搜索框里输入NVIDIA
打开后选择系统信息,点击组件,查看对应的版本信息。 不需要自己去手动安装CUDA和cuDNN,pytorch官方都弄得非常到位了。只需要去官网copy一下命令就行了。
4.创建一个pytorch存放仓库
创建它的原因是将不同版本的pytorch隔开,方便使用。打开anaconda prompt,创建一个叫做mytorch的仓库,也可以叫其它名字 conda create -n mytorch python=3.7
然后输入y 回车
安装完成后输入
conda info --envs 再输入 conda activate mytorch 激活当前环境并进入刚刚创建的仓库
5.安装pytorch
进入pytorch官网 下拉看到该页面,按照自己计算机的版本选择对应的pytorch版本号 我的电脑支持的CUDA版本是11.1.114 进入之前版本号选择 注意,前面没有GPU的这里直接选择CPU即可,不需要再去安装什么CUDA了
在刚才的命令行里输入对应的命令
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge
输入y回车,耐心等待,完成后是这个样子
6.验证
接下来是验证时刻,如下图所示则安装成功
python
import torch
import torchvision
torch.cuda.is_available()
|