坑位:pytorch的版本下载问题
前言
由于实验室的服务器崩掉了,今天重新配之前做强化学习所需要的环境。但是在前一个月很顺利的环境配置过程,今天却卡了壳,一直搞了一晚上才好,特此记录。
问题描述
在装好显卡驱动,cuda,cudnn,pytorch后,深度学习一直无法使用GPU加速,在运行nvidia-smi时,显示无程序占用显卡。
这里我就觉得很奇怪了,毕竟程序也没有报错,应该是在服务器无法调用gpu加速了,只用cpu在进行学习,这样太慢了,显然是不行的。
于是通过命令查看cuda是否可以使用:
命令行输入
python
进入python命令行后,输入
>>>import torch
>>>print(torch