cuda环境安装总结

1.查看自己要安装的cuda版本、对应的driver。对应的网址为:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

如果版本不对应,可以先根据更新显卡驱动到高版本,网址如下:https://www.nvidia.cn/Download/index.aspx?lang=cn

2.下载网址:https://developer.nvidia.com/cuda-toolkit-archive。安装cuda默认安装即可,和cudnn下载网址:https://developer.nvidia.com/rdp/cudnn-archive,二者一定那个要匹配。

3.下载torch是一定要版本对应,可以直接指定。https://download.pytorch.org/whl/cu117

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容