rtx 安装pytorch和cuda
时间: 2023-12-16 19:27:49 浏览: 157
安装cuda以及Pytorch.docx
以下是在RTX上安装PyTorch和CUDA的步骤:
1. 打开命令提示符或Anaconda Prompt(如果已安装Anaconda)。
2. 创建一个新的conda环境并安装PyTorch和CUDA:
```
conda create –n pytorch38 python=3.8 pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
```
这将创建一个名为“pytorch38”的新环境,并在其中安装PyTorch和CUDA。
3. 激活新环境:
```
conda activate pytorch38
```
4. 验证PyTorch和CUDA是否正确安装:
```
python
import torch
print(torch.cuda.is_available())
```
如果输出为True,则表示PyTorch和CUDA已成功安装。
阅读全文