首页 > TAG信息列表 > cu101
cuda10.1GPU版安装PyTroch指令
CUDA 10.1+Cudnn8.0.5 1.7.1版 pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html 1.7.0版 pip install torch==1.7.0+cu101 torchvision==0.8.0+cu101 torchaudio==0.7.0 -f https://dow2021-06-08
搭建ubuntu系统的深度学习环境，自己做个记录 使用的系统环境 使用的系统 ubuntu16.04，之间用20.4的时候跑模型的时候，会出些莫名奇妙的错误。 双硬盘自己的分区情况: efi:主分区,100M /boot:主分区,750M /swap:主分区,32G /:主分区,剩余的固态 /home:逻辑分区,全部的机械硬torch对应的cuda版本及下载链接
torch对应cuda版本 //torch对应cuda版本 torch-1.4.0 torch-1.4.0+cpu torch-1.4.0+cu100 torch-1.4.0+cu101 torch-1.4.0+cu92 torch-1.5.0 torch-1.5.0+cpu torch-1.5.0+cu101 torch-1.5.0+cu102 torch-1.5.0+cu92 torch-1.6.0 torch-1.6.0+cpu torch-1.6.0+cu101 torch-1.torch-scatter -cluster -spline -geometric 安装
今天真的是吐血，一定要把这个分享出去，让更多的人少走弯路。很少有人分享torch-scatter -cluster -spline -geometric 安装，找淘宝店给我安装，他竟然说不会！我只能自己寻找方法了！ 首先，相信很多人看过这个教程：https://github.com/rusty1s/pytorch_geometric#pytorch-170 我来排一排这tensorboard和pytorch分布式卡死有关？
没有去深入研究，只记录现象吧： 依赖Detron2的代码，直接跑分布式卡死，gpu占用率100%完全不动的那种，改成单卡训练后报了tensorboard找不到的错误，于是pip install tensorboard 完成后分布式训练就没问题了。 install之前pip list： Package Version ------------- ----------------