search

thomasbrandon / mish-cuda

Mish Activation Function for PyTorch
MIT License
147 stars 67 forks source link

My server is cuda10.0 installed, and my virtual environment is pytorch1.6 of cuda10.2 installed. #6

Closed xiaozhangya closed 3 years ago

xiaozhangya commented 3 years ago

My server is cuda10.0 installed, and my virtual environment is pytorch1.6 of cuda10.2 installed. In this way, I installed mish-cuda in my virtual environment and reported an error, but failed to install successfully. The error is as follows: 1604472610(1)

thomasbrandon commented 3 years ago

Look at the path it is trying to use to run nvcc. It looks like you've set your CUDA_HOME/CUDA_PATH variables wrongly as that's how PyTorch locates CUDA. As this is in relation to your other issue please reply there rather than creating new issues. Closing this.