
1.安装cuda
(需要nvidia显卡,集成显卡和AMD用不了cuda,直接跳到第二步)

一路默认安装
安装完成后到终端输入 nvidia-smi 验证是否安装成功

2. 安装miniconda
选择自己电脑对应的安装包

一路默认安装
安装完后,进入miniconda的shell看一下安装的python版本是不是我们要的


3.安装GPU版本的PyTorch
(没有Nvidia GPU的在Computer Platform那里选择CPU)
根据电脑选择好后,复制最后一串命令
(注意cuda版本要保持一致,比如你安装了cuda12.x ,这里就不能选择 pytorch-cuda11.x)

复制命令,去miniconda环境里安装即可

然后简单测试一下,能不能在GPU上创建一个张量并进行计算

大功告成