问题:
服务器参数: 3090–ubuntu20.04–cuda10.1–pytorch1.4
由于特殊原因,在服务器上需要使用cuda10.1,在使用pytorch时候,.cuda()指令想把数据转移到GPU上时候,程序会卡死在这里,出不来。
查询一些资料后,发现说是pytorch版本的bug,30显卡基本上使用cuda11往上,但是对于我的工程来说,cuda11不太行,后来通过重新安装pytorch版本解决问题。
解决方案:
安装pytorch1.7.1+cuda11.0 (我的是cuda10.1,但是要安装cuda11.0版本的pytorch)
解决步骤如下:
1、激活anconda中自己的环境
conda activate 自己的空间名字
2、安装pytorch
在上述激活的conda环境下输入下面指令进行安装
Windows和Linux环境下:(以下安装方式选择其一即可)
pip安装:
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
conda安装:
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch
安装成功后打开pycharm,选择自己工程的环境空间,进行测试。成功!
参考链接:
https://blog.csdn.net/qq_33727302/article/details/114839729