Pytorch GPU版本安装（windows10 Anaconda）

最新推荐文章于 2024-04-09 20:42:15 发布

beili6832

最新推荐文章于 2024-04-09 20:42:15 发布

阅读量359

点赞数 1

分类专栏：环境配置

本文链接：https://blog.csdn.net/beili6832/article/details/104998472

版权

2 篇文章 0 订阅

订阅专栏

Pytorch GPU版本安装（windows10）

安装anaconda和CUDA
打开anconda，创建pytorch环境：conda create -n pythorch python=3.7
激活环境：activate pytorch
打开pytorch官网：https://pytorch.org/
get started,如下图：

复制命令，根据自己CUDA版本：conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
上述命令安装pytorch:conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
安装Tensorboard: conda install tensorboard
检查自己是否成功安装pytorch GPU版本：
```
import torch
print(torch.cuda.is_available())
```
返回值为True，代表GPU版本安装成功
根据需要安装opencv
conda install opencv
由于tensorboard是基于早期版本的pillow包，所以可能会报错，我们需要将pillow版本回退到6.1。
conda install pillow=6.1

关注