0
点赞
收藏
分享

微信扫一扫

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

独孤凌雪 2022-04-16 阅读 73

1.创建环境

conda create -n pytorch-gpu python=3.7

2.激活环境

activate pytorch-gpu

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

nvidia-smi

结果如下:
在这里插入图片描述

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

pytorch安装命令
使用该命令进行安装

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

5.测试是否安装成功

在这里插入图片描述
True即为成功安装。

举报

相关推荐

0 条评论