Pytorch——anaconda环境下安装GPU版本
在官网查找版本时,我和 .whl 下载网页进行了对比,因为可能有的版本在 .whl 下载网也中没有,所以我找了相对较新且都能下载的。可能没有对应cuda版本的pytorch,所以即使你版本匹配,也可能会装cpu版的,这就导致测试时,永远是。然后我要安装的anaconda环境下,python版本是3.8的,所以我选择下载。这里为什么要直接对 .whl 文件进行下载,因为我之前修改了清华源下载,而。下载完成后我直接把它们放到了我环境的。首先卸载之前安装的 torch。可以看到,我的CUDA 是 11.1版本。
原创
2023-01-28 10:25:22 ·
850 阅读 ·
0 评论