- 博客(2)
- 收藏
- 关注
RSS订阅原创 浪潮V100显卡CUDA10.1配置pytorch1.7.1的GPU版本
记下我们所需要的版本我们去清华源网站去寻找我们的包，我们只需要找三个，pytorch，torchvison和torchaudio这三个包，我们直接输入后面的版本号在清华源里面寻找，Ctrl+f可以调出网页搜索，找到我们需要的包注意pytorch1.7.1中下面的python版本和cuda版本，不要去安装cpu的就ok，我们使用的是python3.8和CUDA10.1，框出来的就是我需要的版本。在pytorch官网可以看到CUDA版本对应安装的东西，我们CUDA10.1在左下角更多版本去找。
2024-07-06 17:06:33
1074
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人