fairscale库兼容torch1.7.0和cuda11.0的版本安装
时间: 2024-01-14 22:04:34 浏览: 471
CUDA10.1向下兼容10.0,完美运行官方TensorFlow、Pytorch_pytorchcuda10.0,cuda10.0可以升级
5星 · 资源好评率100%
可以通过以下步骤安装fairscale库的兼容版本:
1. 确保已经安装了CUDA 11.0并设置好环境变量;
2. 安装PyTorch版本为torch1.7.0;
3. 运行以下命令安装fairscale库:
```
pip install fairscale
```
4. 如果遇到依赖项的错误,请使用以下命令安装它们:
```
pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.7.0+cu110.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.7.0+cu110.html
pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-1.7.0+cu110.html
pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-1.7.0+cu110.html
```
这样就可以在torch1.7.0和CUDA 11.0上安装兼容的fairscale库了。
阅读全文