python安装gpu-torch
时间: 2023-09-11 09:04:57 浏览: 153
pytorch如何安装
1. 安装CUDA和cuDNN
要安装GPU版本的PyTorch,需要先安装CUDA和cuDNN。请根据你的GPU型号和操作系统版本到NVIDIA官网下载对应的CUDA和cuDNN安装包,并按照安装指南进行安装。
2. 创建虚拟环境
在安装PyTorch之前,我们建议使用conda或者虚拟环境来创建一个独立的Python环境,以避免与其他Python库发生冲突。
首先,安装conda(如果你还没有安装):
```
wget https://repo.anaconda.com/archive/Anaconda3-2021.05-Linux-x86_64.sh
bash Anaconda3-2021.05-Linux-x86_64.sh
source ~/.bashrc
```
然后,创建一个新的conda环境:
```
conda create --name myenv python=3.8
conda activate myenv
```
3. 安装PyTorch
现在可以开始安装PyTorch了。打开终端,输入以下命令安装PyTorch:
```
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia
```
其中,`cudatoolkit=11.1` 是CUDA的版本号,需要根据你的实际安装情况进行修改。
安装完成后,可以在Python中测试是否成功安装:
```python
import torch
print(torch.cuda.is_available())
```
如果输出`True`,说明已经成功安装了GPU版的PyTorch。
阅读全文