cuda版本11.0
torch=1.7.1+cu110
torchvision==0.8.2+cu110
torchaudio==0.7.2
可以直接用命令:
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
也可以去https://download.pytorch.org/whl/torch_stable.html 下载,然后定位,利用文件全名进行安装
同时,tensorflow版本的见网址:(都是CUDA11.0)window10下拯救者笔记本RTX3060laptop配置CUDA11.0 tensorflow-gpu版详解_zhangty1998的博客-CSDN博客windows10RTX3060Laptoppython3.6.5cuda11.0,[cuda11.1(推荐,将bin文件中cusolver64_11.dll,修改为cusolver64_10.dll即可使用,更加方便快捷。)]cudnn-11.0-window