Я установил Pytorch (0.4.1) с CUDA 9.0 через anaconda. Я также установил драйвер Nvidia (390.30). Через nvidia-smi я вижу все графические процессоры (V100) с их ситуацией. Но я всегда получаю False при вызове torch.cuda.is_avaliable (). Я также пытался понизить версию CUDA с 9.0 до 8.0, но ситуация была такой же. Попытался установить Pytorchеще раз, но ничего не изменилось.Кто-нибудь может мне помочь, пожалуйста?
Похоже, это проблема с контейнером Nvidia. После понижения контейнера Nvidia с 18.08 до 18.07 весь код работает нормально.
Лучше позволить Анаконде справиться со всем самостоятельно.В настоящее время у меня есть pytorch версии 0.2, cudatoolkit версии 7.5, cudnn версии 5.1.Работает нормально.Вы можете попробовать
conda update --all
Это должно разрешить все конфликты.