作者:x75066882 | 来源:互联网 | 2023-10-12 12:21
1.序言由于下载了Github的代码,代码运行需要该配置环境RequiredenvironmentUbuntu16.04orabovePyTorch0.4.1oraboveCUDA
1. 序言
由于下载了Github的代码,代码运行需要该配置环境
Required environment
- Ubuntu 16.04 or above
- PyTorch 0.4.1 or above
- CUDA 9.1 or above
但是本机安装的18.04 版本的ubuntu, 所以自己独立开始配置, 在配置过程中,出现了很多问题.
如:1. Nvidia 显卡的算力是 8.6 , 但是安装不同版本的pytorch-cuda 所允许的算例不一样
2. 安装的 pytorch cuda cudnn等版本不匹配
3. 使用了pytorch官网的conda命令行来安装,但是安装的结果显示是cpu版本的pytorch.所以建议使用pip,为了避免错误和麻烦
4. 建议按着该教程以下的步骤,一步步开始
二 查看电脑显卡的算力