安装GPU版本的Pytorch踩坑记录

        别用conda命令安装,就用pip命令安装,前提是安装了对应的CUDA和CUDNN。

        比如现在(2024.12)安装CUDA11.8对应的pytorch官网给出的命令是这个。你的电脑的CUDA可能不是11.8,安装的时候就把这行命令的最后的cu118换成你的CUDA版本。也不用考虑向下兼容不兼容,直接安装符合自己CUDA的版本就ookk。

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

        更正:cuda是向下兼容,pytorch按照cuda版本强适配。

        所以如果你的cuda版本不是pytorch安装命令中的cu***强匹配的,可以降低后面的cu***来安装,比如你的电脑上cuda版本是12.0,pytorch安装命令向下最近只支持cu11.8,那就向下兼容为11.8。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值