Pytorch + CUDA + GPU 的环境搭建

环境依赖一览

• Python 3.9.12
  3 版本基本均可
• NVDIA 517 显卡驱动
  选择自己合适的驱动，显卡 <–适配– 驱动 <–适配– CUDA，显卡版本可以在官网搜索查看，CUDA 和驱动的具体关系可以去官网文档 CUDA Documentation 查看
• CUDA 11.6
  选择自己适合的 CUDA 并下载安装。务必参考上句中的适配性，否则无法正常使用
• Anaconda 22.9.0
  用来便捷安装 pytorch-GPU 和 cudatoolkit 的，pip 由于其非python库的原因无法直接安装 toolkit（但实际上可以用官网所示代码用 --extra-index-url 来安装）
• Pytorch
  使用官网对应命令安装即可，比如 conda install pytorch torc