目录
继极简安装TensorFlow-GPU版本之后,这边继续更新一个极简安装pytorch-gpu教程,依然在Anaconda下面安装,步骤如下:
一、安装pytorch
0、如果没有的话,先去安装个Anaconda、创建虚拟环境
1、打开英伟达控制面板,看看自己的GPU需要什么版本的CUDA,操作如下:
打开控制面板——》NVIDIA控制面板——》系统信息——》组件
比如我的,如图所示,我的版本是CUDA11.0
2、打开pytorch的官网,查看下相应的安装指令
官网地址:https://pytorch.org/get-started/locally/
打开之后如图所示,package部分建议选pip指令安装,感觉下载速度比较快;CUDA部分选上一步中所查到的版本,我的是11.0;然后把那个语句复制一下,我的是:
pip install torch===1.7.1+cu110 torchvision===0.8.2+cu110 torchaudi