Found GPU0 TITAN V which requires CUDA_VERSION >= 9000 for optimal performance and fast startup time

出现如下问题:

Found GPU0 TITAN V 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 http://pytorch.org warnings.warn(incorrect_binary_warn % (d, name, 9000, CUDA_VERSION))
大概意思指的是安装pytorch版本时候编译的cuda版本是8.0,而现在使用的是cuda9.0在调用pytorch,所以造成不匹配。由于我运行的程序使用的是python2.7,pytorch版本<0.4.0,使用pip没法安装pytorch,可尝试使用cuda安装pytorch,执行下列语句。

conda install pytorch=0.3.1 torchvision cuda90 -c pytorch

问题解决。

你可能感兴趣的:(Found GPU0 TITAN V which requires CUDA_VERSION >= 9000 for optimal performance and fast startup time)