1、验证conda是否安装成功
在安装pytorch之前,需要验证coda安装是否成功
WIN+R键,打开cmd,输入nvcc -V
,出现以下界面说明安装成功,并且知道版本为10.2(记住)
2、创建虚拟环境
在命令窗口输入命令cona create -n torch38 python=3.8
,
其中torch38是你创建虚拟环境的名称,可以根据你自己的喜好进行设置,建议不要太复杂。
pthon=3.8是你虚拟环境中python的版本号,也可以根据你自己的需求进行更换。
等待一小会,创建完成后,输入activate torch38
(torch38就是你刚刚创建虚拟环境的名称),进入虚拟环境
3、安装pytorch
进入pytorch官网
点击install,可以看到如下界面
根据我们的需求找到我们需要安装版本对应的命令,但是我们发现,这里不在支持coda10.2。
我们点击上方的install previous versions of PyToch
根据我们自己的版本,找到相应的命令,进行安装即可。如果安装成功,可以直接跳过下面步骤,直接验证自己安装是否成功。
如果安装报错,可以进入命令提供的连接找到自己需要的文件进行本地安装,也可以下载我提供的文件进行本地安装。
首先进入自己存放文件的目录,通过pip install 命令进行安装
前两个文件安装十分顺利,几秒钟就安装好了,但是第三个文件需要安装其他的依赖包,速度明显特别慢,甚至会报错。
这里我们就需要输入命令pip install numpy -i https://pypi.tuna.tsinghua.edu.cn/simple
,安装numpy
然后再安装第三个文件,这样就考可以了
然后依次输入python
,import torch
,torch.cuda.is_available()
,如果返回ture,就证明你安装成功了。
如果你返回false,你可试试以下方法,检查自己安装版本是否正确。一般来说没有问题。
随便用pycharm打开一个项目,按照下图依次点击file->setting->Python Interpreter->齿轮符号->Add
找到你安装虚拟环境的位置,选择python.exe,然后依次点击OK
在这里看看你安装的版本是否正确,如果都是按照前面的步骤进行安装,一般不会出现问题,如果不对,可以重新安装一遍。
4、文件链接
torch百度云链接 提取码:1111
yolov5百度云链接 提取码:1111