win10 + pytorch环境搭建

pytorch环境搭建

1、利用anaconda新建一个pytorch的虚拟环境
去查看https://blog.csdn.net/weixin_43218120/article/details/107708082
2、下载安装，官网的地址为：https://pytorch.org/get-started/locally/，选择适合自己的版本我选择的是torch1.2。
网盘链接：
链接：https://pan.baidu.com/s/1czj4A-uhCyCH20atsynKuw
提取码：t1v8
复制这段内容后打开百度网盘手机App，操作更方便哦
3、测试
（1）在命令行左边为 pytorch 环境中，输入 python
（2）之后，输入 import torch，如果没有报错，意味着 PyTorch 已经顺利安装了
（3）接下来，输入 torch.cuda.is_available，如果是 True，意味着你可以使用 GPU，如果是 False，意味着只能使用CPU。
4、因为我要涉及图像方面的处理，上面的步骤说明pytorch安装完成。不需要的下面步骤不需要再看。
（1）torchvision的安装
安装命令：pip install torchvision
（2）OpenCV-python安装
pip install opencv-python
5、测试
测试代码：

import sys
import time
import cv2
# input = sys.stdin.readline(10)
# print(input)

path = input("请输入文件路径：\n")
print(path)

img = cv2.imread(path)
print(img)
cv2.imshow('picture', img)

cv2.waitKey(0)

# 给出反色图片：

cv2.imshow('picture', 255-img)
cv2.imwrite('fanzhuan.png', 255-img)
cv2.waitKey(0)

cv2.imshow('picture', 66*img)

cv2.imwrite('teshu.png', 66*img)
cv2.waitKey(0)
time.sleep(50)