Pytorch环境搭建
本系列使用vscode的jupyter支持插件进行深度学习开发
vscode下载及安装
可以参考下面这个
vscode下载及安装
安装后需要安装这两个插件即可
Anconda安装–用来创建虚拟环境
可参考下面这个博客
Anconda下载安装
安装好后,打开Anconda Prompt 创建虚拟环境
conda create -n PyTorch python==3.8
打开 PyTorch 官网,根据自己的需求选择
把最下面的代码复制到Anconda Prompt 下运行
验证
在anconda终端栏中依次输入
conda activate PyTorch
python
import torch
torch.cuda.is_available()
CPU:False
GPU: True
为正常
到此环境搭建完毕