网上安装的各种方法,牛鬼蛇神。。我自己记录一个我半个小时就装好的版本。
- 加速系统源
- 安装显卡驱动
- cuda和cudnn
如果是第四步里面的pip
离线安装其实可以忽略安装cuda
和cudnn
,因为显卡驱动默认安装了cuda
对应版本,cuda
可以向下兼容,但是不可以向上兼容!! - 安装pytorch-gpu或者cpu
按我这一套很详细的来,基本上可以了。
网上安装的各种方法,牛鬼蛇神。。我自己记录一个我半个小时就装好的版本。
pip
离线安装其实可以忽略安装cuda
和cudnn
,因为显卡驱动默认安装了cuda
对应版本,cuda
可以向下兼容,但是不可以向上兼容!!按我这一套很详细的来,基本上可以了。