GPU版本pytorch安装过程

GPU版本pytorch安装过程

    • 查看自己电脑支持的CUDA版本
    • 在pytorch官网下载对应的版本
    • 输入语句
    • 查看版本

查看自己电脑支持的CUDA版本

nvidia-smi

GPU版本pytorch安装过程_第1张图片
30系列的显卡一般都是11.3
如果不够11.3的话只能使用10.2版本

在pytorch官网下载对应的版本

https://pytorch.org/
GPU版本pytorch安装过程_第2张图片

输入语句

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
GPU版本pytorch安装过程_第3张图片

查看版本

安装了notebook就可以使用notebook打开查看
GPU版本pytorch安装过程_第4张图片
这样就好了!

你可能感兴趣的:(pytorch,深度学习,python)