PyTorch开发环境配置

关于PyTorch

安装前准备

先阅读官方文档

确定好自己所需要安装的版本以及所需要准备的工具

本文档所使用系统版本为Ubuntu18.04.05LTS

GPU相关参数如图 GPU为 GTX1660ti

[推荐]Anaconda

:-)

Anaconda能很方便的管理虚拟环境，解决各种依赖的冲突

• Anaconda官网

一路安装完成即可

[可选]CUDA

注意

安装CUDA需要NVIDIA的GPU并且安装好显卡驱动

• CUDA

可以根据所选PyTorch对应的CUDA版本进行安装

建议下载本地deb包安装，尽量科学上网，不然可能会因为网络问题下载失败

• 如果出现问题可以查阅官方文档

• CUDNN

根据相应的CUDA版本选择安装

正式安装

根据所准备的选择相对应的选项

conda create -n pytorch-gpu python==3.8
source activate pytorch-gpu
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch