GPU版本——安装pytorch
时间: 2023-11-10 07:04:28 浏览: 119
吴茂贵等人著《Python深度学习——基于PyTorch》书籍代码.zip
好的,针对您的问题,安装 PyTorch GPU 版本的步骤如下:
1. 确认您的电脑已经安装了支持 CUDA 的 NVIDIA 显卡,并且已经安装了 CUDA 工具包和 cuDNN 库。
2. 打开 PyTorch 官网(https://pytorch.org/),选择相应的 PyTorch 版本和 CUDA 版本,复制对应的安装命令。
3. 打开 Anaconda Prompt 或者终端,输入命令并执行安装。
例如,如果您要安装 PyTorch 1.9.0 版本,CUDA 11.1 版本,Python 3.8 版本的 GPU 版本,可以使用以下命令:
```
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia
```
执行完毕后,您就可以在 Python 中使用 PyTorch GPU 版本进行深度学习任务了。
阅读全文