conda安装pytorch gpu 1.6
时间: 2023-10-29 14:32:02 浏览: 153
1. 打开Anaconda Prompt
2. 创建一个新的conda环境:`conda create --name pytorch_gpu_1.6`
3. 激活环境:`conda activate pytorch_gpu_1.6`
4. 安装pytorch:`conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch`
5. 验证安装是否成功:在Python REPL中输入`import torch`,如果没有报错则安装成功。
注意:请确认您的电脑上已经安装了CUDA 10.2,并且您的显卡支持CUDA。
相关问题
window 清华源安装pytorch11.6
要在Window操作系统上安装PyTorch 11.6,首先需要在清华镜像源中下载安装包。在此之前,需要在自己的电脑上安装conda环境,以便使用conda来管理python的安装及PyTorch的包。
第一步,打开清华镜像源的网站(http://mirrors.tuna.tsinghua.edu.cn/),在搜索框中搜索“pytorch 1.6”,选择与自己电脑系统匹配的安装包进行下载。
第二步,打开anaconda prompt或者powershell,输入以下命令创建一个名为“PyTorch”环境:
```
conda create -n PyTorch python=3.7
```
其中,name可以自己定义,python版本需要根据安装包的系统要求进行选择。
第三步,激活创建的环境并安装PyTorch及torchvision,具体命令如下:
```
conda activate PyTorch
conda install pytorch=1.6.0 torchvision cudatoolkit=10.2 -c pytorch
```
其中,cudatoolkit的版本号需要根据自己电脑上的GPU类型确定,如果没有GPU则可以省略该参数。
最后,输入以下命令测试是否安装成功:
```
python
import torch
torch.cuda.is_available()
```
如果输出结果为True,则表明安装完成且成功使用GPU加速。如果输出结果为False,则可能是安装过程中出现了问题,需要进行排查。
阅读全文