- 博客(2)
- 收藏
- 关注
RSS订阅原创 Ubantu系统安装Pytorch以及相应的CUDA
比如说我的CUDA版本是12.2的，那下载cuda11.8或者cuda12.1的都能使用。然后将指令复制，可以把后面的-c pytorch和-c nvidia删掉，这样下载比较快，如果删掉后无法下载，就把它加上。如果返回是Ture的话说明你的pytorch是可以正常调用GPU去进行计算的，如果返回False，说明上面的配置有问题。这里可以根据pytorch官网上对应的版本来安装CUDA的版本，也可以先安装CUDA再选择pytorch的版本。选择刚刚推荐的版本（不用选server的），然后点击应用更改。
2024-08-03 11:32:27
3461
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人