查看CUDA的版本
- nvidia-smi 命令可用于显示 NVIDIA GPU 的状态和信息
- CUDA（Compute Unified Device Architecture） 是由 NVIDIA 开发的一个并行计算平台和编程模型，专为 NVIDIA 图形处理单元（GPU）设计。CUDA 允许开发者利用 GPU 的强大计算能力来加速计算密集型的任务。
- nvcc -v 可以查看 CUDA 的版本
配置 Conda 环境
如果使用的 fish shell 需要初始化 shell
安装 PyTorch
- 地址：pytorch.org/get-started…
- 需要根据支持的 CUDA 安装不同的 PyTorch，如果不能使用 CUDA 就选择 CPU only
- 根据 CUDA 版本选择合适的 pytorch 执行命令安装
- 带有 CUDA 的 PyTorch
VSCode 配置 Python 环境
- 需要安装的插件
- 如果是 windows 可以将bash 修改为 Git Bash
