新显卡用了两个月第一次想做点正事,结果...

! macro-avatar

! macro

2020-12-03T15:00:21+00:00

[img]https://img.nga.178.com/attachments/mon_202012/04/9aQ5-jv3jKsT3cStr-4g.jpg[/img]
找到了解决办法是用不稳定的nightly版本,但是只是能用,性能大打折扣还有各种不明issue。
conda install pytorch torchvision cudatoolkit=11 -c pytorch-nightly

折腾了一下,最后只能用笔记本...
RiverDragonz-avatar

RiverDragonz

试试
nvcr.io/nvidia/pytorch:20.11-py3

明确表示支持安培
! macro-avatar

! macro

Reply to [pid=473769398,24483424,1]Reply[/pid] Post by [uid=39414951]充氡气泡水[/uid] (2020-12-04 23:10)
一样啊,这个镜像里面的是pytorch1.8就是基于之前的nightly的