查看GPU
nvidia-smi
我的CUDA Version:10.1,比较低,没法按照官网上的使用,但是只要不涉及到GPU还是能用其它功能的嘞。
python版本
python -V
安装Pytroch
在官网上下载【PyTorch】
选择【Get Started】也就是开始,在菜单的第一个位置。
匹配版本
我这里CUDA是10.1的,但是没有,所以只能选择11.3的CUDA。选择完毕后可以看到【Run this Command】的提示语句。
安装PyTorch
我这里分开安装了,先安装pytorch再安装torchaudio
conda create -name pytorch python=3.9
比较大,需要安装个半小时左右。
我这里安装的时候名字写错了,搞成了【ame】
切换一下环境
conda activate ame
查看安装环境,这里1.12.1就是咱们刚才安装的。
安装【torchaudio】
conda install torchaudio cudatoolkit=11.3
安装完成后通过【pip list】查询一下
一大堆,OK,安装完毕。能用了。
我这个版本1.12.1,需要根据自己的需求进行该版本。