![](https://fly-uni.com/wp-content/uploads/2023/10/image-79-1024x329.png)
1.安装cuda
(需要nvidia显卡,集成显卡和AMD用不了cuda,直接跳到第二步)
![](https://fly-uni.com/wp-content/uploads/2023/10/image-70-1024x628.png)
一路默认安装
安装完成后到终端输入 nvidia-smi 验证是否安装成功
![](https://fly-uni.com/wp-content/uploads/2023/10/image-71-1024x325.png)
2. 安装miniconda
选择自己电脑对应的安装包
![](https://fly-uni.com/wp-content/uploads/2023/10/image-72-1024x612.png)
一路默认安装
安装完后,进入miniconda的shell看一下安装的python版本是不是我们要的
![](https://fly-uni.com/wp-content/uploads/2023/10/image-73.png)
![](https://fly-uni.com/wp-content/uploads/2023/10/image-74.png)
3.安装GPU版本的PyTorch
(没有Nvidia GPU的在Computer Platform那里选择CPU)
根据电脑选择好后,复制最后一串命令
(注意cuda版本要保持一致,比如你安装了cuda12.x ,这里就不能选择 pytorch-cuda11.x)
![](https://fly-uni.com/wp-content/uploads/2023/10/image-76-1024x725.png)
复制命令,去miniconda环境里安装即可
![](https://fly-uni.com/wp-content/uploads/2023/10/image-77-1024x136.png)
然后简单测试一下,能不能在GPU上创建一个张量并进行计算
![](https://fly-uni.com/wp-content/uploads/2023/10/image-78.png)
大功告成