环境搭建

AI的环境搭建是一件稍有门槛的事情,这关系到操作系统、包、编程语言、CUDA版本等等,所以在安装之前请对包括但不限于 Python,pip,CUDA,CUDNN 的内容有一定的了解。同时也极力推荐大家使用虚拟环境安装

PyTorch 的安装

请在 PyTorch 官网 中找到下表并选择自己的操作系统、包、编程语言、CUDA版本,随后运行给出的命令。

PyTorch 和 CUDA 的安装

TensorFlow 的安装

相较于 PyTorchTensorFlow 的安装就复杂许多了,以下给出一些官方的网站、文档。

最后更新于