硬件: 显卡(GPU) 8G以上

我使用的是NVIDIA RTX A4000 GPU 16G

环境配置:

1. 安装显卡驱动:官方驱动 | NVIDIA

根据自己的显卡搜索对应的驱动

2. 安装CUDA:CUDA Toolkit Archive | NVIDIA Developer

CADA,cuDNN以及Torch需要版本对应

目前使用的是

3. cuDNN安装:CUDA Deep Neural Network (cuDNN) | NVIDIA Developer

需要注册后登陆,选择对应CUDA11.7.1的cuDNN

4. 安装anaconda:Anaconda | The World's Most Popular Data Science Platform

下载windows版本,目前集成了python3.9

5. 安装Torch:Start Locally | PyTorch

可以选择conda安装或pip安装,注意CUDA版本

6. 可选,安装pycharm:PyCharm: the Python IDE for Professional Developers by JetBrains

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐