3090显卡服务器使用torch中的.cuda()语句会卡死问题解决方法

问题：

服务器参数： 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