在Anaconda prompt终端的基本环境中,安装Pytorch(GPU版)的指令
时间: 2023-12-03 22:44:09 浏览: 186
安装Pytorch(GPU版)的指令如下:
```
conda install pytorch torchvision torchaudio cudatoolkit=X.X -c pytorch
```
其中,`X.X`是对应的CUDA版本号,需要根据自己的环境来选择。如果不知道CUDA版本号,可以使用以下指令查询:
```
nvcc --version
```
相关问题
anaconda prompt输入什么指令来安装pytorch1.6.0与cuda10.1
可以使用以下指令来安装PyTorch 1.6.0和CUDA 10.1:
```
conda install pytorch=1.6.0 torchvision=0.7.0 cudatoolkit=10.1 -c pytorch
```
请注意,这个指令需要在Anaconda Prompt中执行,并且需要已经安装了CUDA 10.1。如果您的系统没有安装CUDA 10.1,您需要先安装CUDA 10.1并配置好环境变量,再执行上述指令。
在anaconda prompt中输入什么指令来安装pytorch1.6.0
在anaconda prompt中输入以下指令来安装pytorch1.6.0:
```
conda install pytorch=1.6.0 torchvision cudatoolkit=10.2 -c pytorch
```
请注意,这需要在使用NVIDIA GPU的计算机上安装CUDA Toolkit 10.2。如果您没有GPU或不想使用它,可以使用以下指令安装CPU版本的pytorch:
```
conda install pytorch-cpu=1.6.0 torchvision-cpu -c pytorch
```
阅读全文