Pytorch 环境搭建
导航
Pytorch 环境搭建¶
环境配置:学习 Anaconda3 教程,然后配置 PyTorch。
本地服务器支持 GPU¶
替换 本地服务器 GPU 配置 中的环境
ai
为torch
。安装 PyTorch GPU 支持:
conda install pytorch torchvision torchaudio -c pytorch
添加 Jupyter 内核支持:
conda install ipykernel python -m ipykernel install --user --name torch --display-name "PyTorch"
更多内容见 预览:pytorch-demo。