但有的对应安装包无法使用,有可能是由于卡太旧的原因。步骤一:使用nvidia-smi查询驱动版本:如图中Driver Version所示,该卡目前的驱动版本为384.81。此方法指定PyTorch版本后,conda会自动匹配到合适版本的CUDA Toolkit。运行conda install pytorch=1.1.0 cudatoolkit=9.0 -c pytorch即可。

解决PyTorch与CUDA版本不匹配的问题

解决PyTorch与CUDA版本不匹配的问题

解决PyTorch与CUDA版本不匹配的问题