①查看cuda toolkit版本命令(在Win\linux\Max均可):
nvcc --version
结果如下:
②找到pytorch对应的版本并安装
因为①中cuda版本为11.7,可将命令修改为
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
注意:
cuda、python、pytorch的版本一定要注意对应,否则运行中会遇到各类错误。因为版本不匹配导致的Error,最好卸载环境(可以使用虚拟环境)后重新安装。
③有时作为Linux普通用户,我们无法获得root权限,无法安装各类工具包,甚至无法安装pip3,这时可以配置conda虚拟环境
由于Anconda官网下载较慢,可以使用清华大学开源软件镜像站下载
找到适合自己系统的版本,例如linux,是后缀名为.sh的安装文件,下载完成后找到对应的目录,运行
bash Anaconda3-2023.09-0-Linux-x86_64.sh
安装完成后重启终端
随后,创建自己的虚拟环境(一个好习惯,对于不同项目可以配置不同的环境,以免版本冲突)
#创建虚拟环境
conda create -n neural_object_fitting python=3.7
#激活创建的虚拟环境
conda activate neural_object_fitting
#安装pytorch
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
#安装其他依赖库
pip install -r requirements.txt
④有时我们发现pip install下载很慢,时常有Time Out的错误,这时可以用国内镜像。
常用的有:
#豆瓣
https://pypi.douban.com/simple
#阿里云
https://mirrors.aliyun.com/pypi/simple
#清华大学
https://pypi.tuna.tsinghua.edu.cn/simple
#中国科技大学
https://pypi.mirrors.ustc.edu.cn/simple
具体使用方法:
#以豆瓣为例
pip install -r requirements.txt -i https://pypi.douban.com/simple
⑤工位vscode通过ssh远程连接实验室服务器,在断电开机的情况下,将电脑带回去。
本以为ssh断开连接后,程序必须重新运行。但是实际上,在重新连接服务器,reload窗口后,代码竟然继续运行。
个人猜测是因为没有退出vscode,终端还是开着的。若退出vscode,那就需要真的重新运行代码了,什么时候测试一下。