11.6cuda 安装pytorch命令
时间: 2023-05-24 21:06:53 浏览: 364
深度学习环境配置Anaconda+Pycharm+CUDA+CUdnn+PyTorch+Tensorflow.txt
在安装 PyTorch之前,您需要使用conda创建一个新环境。然后,使用以下命令安装PyTorch:
```
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia
```
其中`cudatoolkit`版本号应与您的 CUDA 版本相对应。您可以在安装步骤之前运行以下命令来检查 CUDA 版本:
```
nvcc -V
```
阅读全文