Pytorch配置环境
cuda与cudnn的下载
cudatoolkit下载:https://developer.nvidia.com/cuda-toolkit-archive
cudnn下载:https://developer.nvidia.com/rdp/cudnn-archive
cuda支持情况:https://developer.nvidia.com/cuda-gpus#compute
cuda和cudnn的对应关系:https://developer.nvidia.com/rdp/cudnn-archive#a-collapse742-10
使用 PyTorch 查看 CUDA 和 cuDNN 版本
import torch
print(torch.__version__)
print(torch.version.cuda)
print(torch.backends.cudnn.version())
安装命令
配置清华源码
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
安装命令
pip install torch==1.10.1+cu102 torchvision==0.11.2+cu102 torchaudio==0.10.1+cu102 -f https://download.pytorch.org/whl/torch_stable.html
版本对应:文章来源:https://www.toymoban.com/news/detail-554285.html
torch
与torchvision
对应关系,来源:官方Repo: vision
torch | torchvision | python |
---|---|---|
main / nightly | main / nightly | >=3.8, <=3.10 |
2.0.0 | 0.15.1 | >=3.8, <=3.10 |
1.13.0 | 0.14.0 | >=3.7.2, <=3.10 |
1.12.0 | 0.13.0 | >=3.7, <=3.10 |
1.11.0 | 0.12.3 | >=3.7, <=3.10 |
1.10.2 | 0.11.3 | >=3.6, <=3.9 |
1.10.1 | 0.11.2 | >=3.6, <=3.9 |
1.10.0 | 0.11.1 | >=3.6, <=3.9 |
1.9.1 | 0.10.1 | >=3.6, <=3.9 |
1.9.0 | 0.10.0 | >=3.6, <=3.9 |
1.8.2 | 0.9.2 | >=3.6, <=3.9 |
1.8.1 | 0.9.1 | >=3.6, <=3.9 |
1.8.0 | 0.9.0 | >=3.6, <=3.9 |
1.7.1 | 0.8.2 | >=3.6, <=3.9 |
1.7.0 | 0.8.1 | >=3.6, <=3.8 |
1.7.0 | 0.8.0 | >=3.6, <=3.8 |
1.6.0 | 0.7.0 | >=3.6, <=3.8 |
1.5.1 | 0.6.1 | >=3.5, <=3.8 |
1.5.0 | 0.6.0 | >=3.5, <=3.8 |
1.4.0 | 0.5.0 | ==2.7, >=3.5, <=3.8 |
1.3.1 | 0.4.2 | ==2.7, >=3.5, <=3.7 |
1.3.0 | 0.4.1 | ==2.7, >=3.5, <=3.7 |
1.2.0 | 0.4.0 | ==2.7, >=3.5, <=3.7 |
1.1.0 | 0.3.0 | ==2.7, >=3.5, <=3.7 |
<=1.0.1 | 0.2.2 | ==2.7, >=3.5, <=3.7 |
torch
与torchaudio
对应关系,来源:官方Repo: audio
torch | torchaudio | python |
---|---|---|
main / nightly | main / nightly | >=3.8, <=3.10 |
2.0.1 | 2.0.2 | >=3.8, <=3.11 |
2.0.0 | 2.0.1 | >=3.8, <=3.11 |
1.13.1 | 0.13.1 | >=3.7, <=3.10 |
1.13.0 | 0.13.0 | >=3.7, <=3.10 |
1.12.0 | 0.12.0 | >=3.7, <=3.10 |
1.11.0 | 0.11.0 | >=3.7, <=3.9 |
1.10.0 | 0.10.0 | >=3.6, <=3.9 |
1.9.1 | 0.9.1 | >=3.6, <=3.9 |
1.9.0 | 0.9.0 | >=3.6, <=3.9 |
1.8.2 | 0.8.2 | >=3.6, <=3.9 |
1.8.1 | 0.8.1 | >=3.6, <=3.9 |
1.8.0 | 0.8.0 | >=3.6, <=3.9 |
1.7.1 | 0.7.2 | >=3.6, <=3.9 |
1.7.0 | 0.7.0 | >=3.6, <=3.8 |
1.6.0 | 0.6.0 | >=3.6, <=3.8 |
1.5.0 | 0.5.0 | >=3.5, <=3.8 |
1.4.0 | 0.4.0 | ==2.7, >=3.5, <=3.8 |
- 当已经装好
torch
包时,pip install torchvision torchaudio
会自动寻找对应的版本安装。 等待再次验证是否有效。
备注
帮忙安装各种深度学习环境:anaconda,pycharm,tensorflow,pytorch,虚拟机配置等等,也可帮忙配置和复现顶会论文;可联系1505413361@qq.com,不是free的哈,伸手党勿扰谢谢文章来源地址https://www.toymoban.com/news/detail-554285.html
到了这里,关于Pytorch环境配置的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!