1. 首先需要先查看nvcc的位置which nvcc2. 查看现使用的cuda版本。或者如下。相信大部分同学conda创建环境是为了解决cuda和torch的对应关系问题。此时有一定概率，两者显示的cuda版本不一样。如果不影响后续开发，可以以import torch时显示的版本为准继续后续工作。如果你和我一样安装其他库时，报了cuda版本不对应的问题，就继续往下看吧！

这里采用的是新版版本对应很重要，尝了一下onnxruntime-gpu最新版1.60搭配cuda1.7，文末给测试搭建是否成功的简易代码。

这里采用的是新版版本对应很重要，尝了一下onnxruntime-gpu最新版1.60搭配cuda1.7，文末给测试搭建是否成功的简易代码。

【代码】服务器指定显卡。

linux服务器安装pytorch及版本对应