1. 首页
  2. 编程语言
  3. Python
  4. 解决PyTorch与CUDA版本不匹配的问题

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

上传者: 2022-01-03 13:46:00上传 PDF文件 163.97 KB 热度 27次

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

用户评论