1.配置pytoch环境需要cuda时:
(1)在pytorch官网:PyTorch选择自己需要的版本下载
复制下面的链接即可pip安装
(2)需要特定的版本,例如需要 pytorch==1.9.1==py3.6_cuda11.1_cudnn8.0.5_0的版本,用以下语句安装
pip install torch==1.9.1+cu111 torchvision==0.10.1 torhaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html
(3)conda安装,可先到Index of /anaconda/cloud/pytorch/linux-64/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror网站上下载源:pytorch-1.8.0-py3.8_cuda11.1_cudnn8.0.5_0.tar.bz2
torchvision-0.9.0-py38_cu111.tar.bz2
然后输入:conda install 下载的文件
2.安装很慢的解决方法
(1)conda网速不稳定,安装很慢,可用mamba加速:GitHub - mamba-org/mamba: The Fast Cross-Platform Package Manager
(2)pip安装时下载很慢,可更改源
临时使用
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple some-package
注意,simple 不能少, 是 https 而不是 http
设为默认
升级 pip 到最新的版本 (>=10.0.0) 后进行配置:
python -m pip install --upgrade pip pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
如果您到 pip 默认源的网络连接较差,临时使用本镜像站来升级 pip:
python -m pip install -i https://pypi.tuna.tsinghua.edu.cn/simple --upgrade pip



