Python版本
Win10上的Python 3.10.8
Anaconda采用2022版,自带python3.9
先看一下,我本人CUDA的版本
然后我们打开pytorch官网
传送门来咯:Start Locally | PyTorch
惊喜的发现有适合我的版本,我选择pip方式安装QAQ
如果没有适合你的版本的话,请点击箭头方向的东西
然后打开Anaconda Prompt,复制官网上的命令,然后就可以下载了!!!
图为博主已经安装过的(迹象)
Pytorch GPU版本安装
我的CUDA是11.6版本的(看第三张图),直接复制官网上的conda命令就可以辣!!! tips:如果你还是找不到适合的版本,假如你的CUDA是11.4的,那你安装1.9.0的,安装降低版本的!!
安装后,我们来测试一下~~~~
打开我的vscode,输入一个5行2列的矩阵,看看能不能打印出来(这也是官方上的一个小例子)
现在看看GPU版本有没有安装好,显示为True,就代表安装好了