问题
深度学习程序,在服务器运行,需要借助GPU加速。为了检测是否开启了GPU加速,采用以下代码:
~python
>> import torch
>> torch.cuda.is_available()
>> false #说明没有使用GPU加速
安装过程
安装老版本的pytorch官方网站
搭配CUDA 11.1版本
命令行统一安装(网速限制,不建议使用)
# CUDA 11.1
pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
离线下载进行安装(对应上面命令,一共三个whl后缀文件)
去官网下载适配本机的cuda=11.1的torch(https://download.pytorch.org/whl/torch_stable.html)
torch==1.9.0+cu111:
解释名词含义:
cu111: 表示cuda版本为11.1
torch-1.9.0::表示torch版本为1.9.0
**%2B:表示命令里面的 + 号
cp37: 表示使用python版本为3.7
linux:表示适用于linux系统
x86_64:表示同时兼容32和64位系统
torchvision==0.10.0+cu111:
torchaudio==0.9.0:
最后安装三个whl后缀的文件
pip install 下载的文件所在目录/文件名
可以检验一下是否安装成功,进入python环境,输入如下命令:
import torch
print(torch.cuda.is_available())
如果输出是true,就代表安装成功。