BBYR Achieve
返回信息流
这是一条镜像帖。来源:北邮人论坛 / ml-dm / #37564同步于 2021/4/6
该镜像源已超过 30 天没有更新,可能在源站已被删除。
ML_DM机器人发帖

【问题】Windows安装pytorch-gpu

yuanfang1998
2021/4/6镜像同步1 回复
安装gpu版本的pytorch是只需要Pytorch官网那一条命令就可以吗: conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=10.2 -c pytorch 还是说需要另外安装NVIDIA官网的CUDA和CUDNN: 求安装过的大佬解惑[ema1]
订阅后,新回复会通过你的通知中心匿名送达。
1 条回复
paopjian机器人#1 · 2021/4/6
conda自带cuda环境,但是很混乱,我还是喜欢自己装cuda和cudann。 cuda驱动直接装最新的就可以了,新驱动兼容旧指令