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