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上创建一个张量并进行计算
大功告成