我已经安装了cuda,通过nvcc -V查询我的cuda版本是
cuda版本是11.2
接下来,我执行了下面命令后,环境配置成功了(python是3.8)
pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html --trusted-host pypi.org --trusted-host download.pytorch.org --trusted-host files.pythonhosted.org
————————————————
版权声明:本文为CSDN博主「m0_58709899」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/m0_58709899/article/details/132716515