Unable to Install cu90 Pytorch?

1

Dear all:
I found I have a big problem in install cu90 version of pytorch:

/home/llq/conda/lib/python3.6/site-packages/torch/cuda/__init__.py:115: UserWarning: 
    Found GPU0 Tesla V100-SXM2-16GB which requires CUDA_VERSION >= 9000 for
     optimal performance and fast startup time, but your PyTorch was compiled
     with CUDA_VERSION 8000. Please install the correct PyTorch binary
     using instructions from https://pytorch.org
    
  warnings.warn(incorrect_binary_warn % (d, name, 9000, CUDA_VERSION))

I tried several ways to install such as:

conda install pytorch torchvision -c pytorch
conda install pytorch torchvision cuda90 -c pytorch
pip install torch....whl

No mater i try, the system will install cu80 for me.
I verified my pc installed cuda90 via nvcc -V.

2

Hi,

I would expect that you have an old cuda8.0 pytorch install lingering somewhere that is being picked up instead of what you install.
Make sure to uninstall all pytorch / pytorch-nightly packages before installing the new one.