Pytorch 深度学习开发环境搭建流程

146 阅读1分钟

1. 安装 Anaconda 并创建环境

在官网按步骤安装就好,安装后使用如下命令创建和激活环境

conda create -n xxx

conda activate xxx

Tips

  • 如何退出 python 命令行?

    exit() / Ctrl + D

  • 如何查看 conda 版本?

    不用进入 python 命令行,直接 conda -V 就好

2. 安装 Pytorch

nvidia-smi 查看驱动版本

image.png

对应下图,下载 Pytorch 时 CUDA 可以选择 11.8 及以下

image.png

image.png

选好各项后复制命令安装,conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia,要下 2.35G 的内容

image.png

3. 开发环境

VSCode 连接远程服务器没有代码提示

按顺序卸载 Jupyter 和 Python 后,重新安装就好

image.png

torch 没有代码提示

文件 - 首选项 - 设置 中 设置 Jdei Server

image.png