ubuntu18.04安装GPU PyTorch

最新推荐文章于 2024-07-06 19:17:17 发布

qq_41549249

最新推荐文章于 2024-07-06 19:17:17 发布

阅读量520

点赞数 1

文章标签： pytorch ubuntu gpu

原文链接：https://blog.csdn.net/lovely_yoshino/article/details/107142183?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522160490801319724835846096%2522%252C%2522scm%2522%253A%252220140713.130102334.pc%255Fall.%2522%257D&request_id=160490801319724835846096&bi

版权

转载自这篇文章

安装GPU版本的PyTorch

这里选择用pip进行安装，首先需要安装pip：执行命令sudo apt intall python-pip3

（该步骤可以跳过） 现在建议配置pip虚拟环境，为此我们需要配置virtualenv（有关virtualenv的详细介绍可以看看廖雪峰Python3教程和官方documentation）：

安装virtualenv：执行命令pip3 install virtualenv
新建一个虚拟环境（这里取名为env-pytorch)：在终端中切换到想要新建env-pytorch的目录下（创建完虚拟环境后此目录下会多一个名为env-pytorch的文件夹，包含该虚拟环境相关的全部文件），执行命令virtualenv --no-site-packages venv -p <PYTHON_EXE>，其中<PYTHON_EXE>是Python的版本号，例如python3.6
到官网选择对应版本之后会得到相应的安装命令，如下图：

可以通过下面的命令进行测试：

python					#进入python
import torch
import torchvision

#返回当前设备索引
torch.cuda.current_device()

#返回GPU的数量
torch.cuda.device_count()

#返回gpu名字，设备索引默认从0开始
torch.cuda.get_device_name(0)

#cuda是否可用
torch.cuda.is_available()