Ubuntu下GPU版本pytorch安装及虚拟环境配置

1.创建环境

conda create -n pytorch-gpu python=3.7

2.激活环境

activate pytorch-gpu

3.在激活的环境中使用命令查看cuda的版本

nvidia-smi

结果如下：

4.在pytorch官网（https://pytorch.org/）查找对应版本的pytorch安装命令进行安装

使用该命令进行安装

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

5.测试是否安装成功

True即为成功安装。