VScode 运行 jupyter 心得

在服务器上，用 vscode 运行 .ipynb 文件是常用的手段，但是搞多了就会发现还是会有各种问题，在这里记录一下。

1. os.environ 的使用

经常在这个运行一个程序前，我们需要加载一些环境变量，来设置比如代理转发（用于下载外面的东西），例：

import os
# 修改 huggingface 下载源为镜像源
os.environ['HF_ENDPOINT'] = 'https://hf-mirror.com'

# 修改网络代理端口
proxy_list = ['HTTP_PROXY', 'HTTPS_PROXY', 'http_proxy', 'https_proxy']
for proxy in proxy_list:
    os.environ[proxy] = 'http://u-cEoRwn:EDvFuZTe@172.16.4.9:3128'
# 注：不同服务器后面设置的内容不一样，自己改改。

也可以用这个去指定 CUDA_VISIBLE_DEVICES，如

os.environ['CUDA_VISIBLE_DEVICES'] = '4,5,6,7'

但是注意，环境变量的设置最好在最开始，即代码开始的代码设置，再 import 其它包，否则就可能不生效了。

2. 手动启动

有时候环境加载时非常复杂的，不一定说设置一些环境变量就行，比如我现在的服务器是需要手动加载 cuda 的，这个时候就只能先加载环境再手动启动了。

如 source /home/bingxing2/apps/package/pytorch/2.1.0+cuda118_cp310/env.sh 这样加载，

那只能加载完这个和虚拟环境以后，启动

jupyter notebook

记录端口，例：

http://127.0.0.1:8890/?token=cfb7d47e2a70f09f8997f24b09995dbb147c9467c051d188

然后 Vscode 打开文件，在右上角点一下，选择其它内核，现有 jupyter 服务器，输入这个即可。