CUDA11.2 RTX3080 pytorch1.9.0 安装
Pytorch官网: https://pytorch.org//get-started/locally/.
没有CUDA11.2怎么办?
选择Previous Pytorch Versions,发现也没有CUDA11.2,于是安装了CUDA11.1对应的pytorch版本,选择了1.9.0,跑模型没什么毛病。
pip install torch==1.9.0+cu111 torchvision==
Pytorch官网: https://pytorch.org//get-started/locally/.
没有CUDA11.2怎么办?
选择Previous Pytorch Versions,发现也没有CUDA11.2,于是安装了CUDA11.1对应的pytorch版本,选择了1.9.0,跑模型没什么毛病。
pip install torch==1.9.0+cu111 torchvision==