为了方便,记录一下本次pytorch的搭建,主要参考网上的一些博客帖子,一些引用也会记录出处,本贴仅用于个人学习。
确定需要搭建的环境
anaconda+python3.7+pytorch1.13.1
从NVIDIA控制面板-帮助-系统信息-组件可查看cuda的版本号(9.2)
python可以在python.org/downloads/进行下载,由于github里面的要求是python3.7,所以本次利用prompt来创建python3.7的环境。
conda create -n py37_torch131 python=3.7
在安装pytorch之前还需要先装一个numpy,这次选择(numpy-1.18.4-cp37-cp37m-win_amd64.whl)这个版本,下载地址为https://pypi.org/project/numpy/#files。
然后在https://pytorch.org/get-started/locally/#mac-python可以查看到所需要的pytorch的版本(torch-1.3.1+cu92-cp37-cp37m-win_amd64.whl)并进行安装。
主要是因为直接在prompt中用conda命令进行安装经常失败,所以还是直接把包下下来然后使用本地包进行安装会比较好一点。
其他还有一些包需要安装一下:
pip install jupyter tqdm opencv-python matplotlib pandas
当然这么直接装肯定是会失败的,所以还是一步一步来吧,从网上把包下下来,然后pip install本地的包。
opencv_python-4.2.0.34-cp37-cp37m-win_amd64.whl https://pypi.org/project/opencv-python/#files
matplotlib-3.2.1-cp37-cp37m-win_amd64.whl https://pypi.org/project/matplotlib/#files
pandas-1.0.3-cp37-cp37m-win_amd64.whl https://pypi.org/project/pandas/#files
会发现在安装过程中会有很多的小的包报错,不用担心,同理直接上网找到那些网站下载本地文件再进行pip就可以了,因为实在太多,所以就不一一列举。
在下一步进行运行的时候发现没有装torchvision以及cudatoolkit所以再进行安装,直接运行conda代码
conda install cudnn=7.6.0
会发现清华源报错,找不到相关的文件,所以把清华源都删了,直接用默认的channel进行conda安装。
然后就是torchvision了,这个花了一上午的时间终于搞定,首先是版本匹配的问题,然后找到了一个类似的,只是版本不同,多的不说,上代码。
pip install torchvision===0.4.2 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple
说明
这是我第一次发帖,如果有不对的地方希望大家能够指正一下,共同进步。
参考
https://www.jianshu.com/p/468b23f38ea6
https://damagecontrol.studio/Conda-cai-keng/
https://github.com/datawhalechina/team-learning/blob/master/