masahi commented on pull request #10758:
URL: https://github.com/apache/tvm/pull/10758#issuecomment-1086294177


   When we install torch via `pip install torch==1.11.0`, how do we know which 
cuda it was built against? Maybe we should explicitly specify the cuda version 
as in https://pytorch.org/ (`pip3 install torch -extra-index-url 
https://download.pytorch.org/whl/cu113`). But a better way is still to install 
a CPU-only version.
   
   Or we can try updating our NVIDIA container image to update cuda itself (we 
should do that anyway sooner or later). I don't know what process is required 
for that.
   
   


-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: [email protected]

For queries about this service, please contact Infrastructure at:
[email protected]


Reply via email to