如何高效安装pytorch_cuda_gpu【一起来动手学深度学习啦】

148 阅读1分钟

Python版本

Win10上的Python 3.10.8

Anaconda采用2022版,自带python3.9

先看一下,我本人CUDA的版本

image.png

image.png

image.png 然后我们打开pytorch官网 传送门来咯:Start Locally | PyTorch

image.png 惊喜的发现有适合我的版本,我选择pip方式安装QAQ

image.png 如果没有适合你的版本的话,请点击箭头方向的东西

image.png 然后打开Anaconda Prompt,复制官网上的命令,然后就可以下载了!!!

image.png 图为博主已经安装过的(迹象)

image.png

Pytorch GPU版本安装

我的CUDA是11.6版本的(看第三张图),直接复制官网上的conda命令就可以辣!!! tips:如果你还是找不到适合的版本,假如你的CUDA是11.4的,那你安装1.9.0的,安装降低版本的!!

image.png

image.png 安装后,我们来测试一下~~~~
打开我的vscode,输入一个5行2列的矩阵,看看能不能打印出来(这也是官方上的一个小例子)

image.png 现在看看GPU版本有没有安装好,显示为True,就代表安装好了

image.png